conceptofmind / vit-flaxLinks
Implementation of numerous Vision Transformers in Google's JAX and Flax.
☆22Updated 2 years ago
Alternatives and similar repositories for vit-flax
Users that are interested in vit-flax are comparing it to the libraries listed below
Sorting:
- FID computation in Jax/Flax.☆27Updated 10 months ago
- The 2D discrete wavelet transform for JAX☆43Updated 2 years ago
- This is a port of Mistral-7B model in JAX☆32Updated 11 months ago
- Automatically take good care of your preemptible TPUs☆36Updated 2 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 4 years ago
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- Utilities for PyTorch distributed☆24Updated 3 months ago
- PyTorch interface for TrueGrad Optimizers☆42Updated last year
- Implementation of Vision Transformers in Flax☆18Updated 4 years ago
- A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…☆27Updated last month
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated 2 years ago
- ☆29Updated 6 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆81Updated last year
- A metrics library for the JAX ecosystem☆40Updated 2 years ago
- LoRA for arbitrary JAX models and functions☆136Updated last year
- JAX Implementation of Black Forest Labs' Flux.1 family of models☆33Updated 7 months ago
- Experiment of using Tangent to autodiff triton☆79Updated last year
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆20Updated last year
- Unofficial JAX implementations of deep learning research papers☆156Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆88Updated 11 months ago
- Lightning-like training API for JAX with Flax☆38Updated 5 months ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 7 months ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆58Updated 3 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆58Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆110Updated 3 years ago
- Train vision models using JAX and 🤗 transformers☆97Updated last month
- A simple Transformer where the softmax has been replaced with normalization☆20Updated 4 years ago