Alternatives and similar repositories for vit-flax:

Users that are interested in vit-flax are comparing it to the libraries listed below

• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 7 months ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 4 months ago
• lucidrains / local-attention-flax
  Local Attention - Flax module for Jax
  ☆20Updated 3 years ago
• vballoli / vit-flax
  Implementation of Vision Transformers in Flax
  ☆18Updated 4 years ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆120Updated 7 months ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆23Updated last month
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆17Updated last year
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated 2 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆147Updated 2 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆75Updated 4 months ago
• FrancescoSaverioZuppichini / Loading-huge-PyTorch-models-with-linear-memory-consumption
  Little article showing how to load pytorch's models with linear memory consumption
  ☆34Updated 2 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 8 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆40Updated 2 years ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆86Updated last year
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆96Updated last month
• radarFudan / mamba-minimal-jax
  ☆30Updated 3 months ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 3 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆85Updated 6 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 7 months ago
• peerdavid / layerwise-batch-entropy
  Layerwise Batch Entropy Regularization
  ☆22Updated 2 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated this week
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆57Updated last year
• HomebrewML / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆42Updated last year
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 3 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆222Updated 2 weeks ago