conceptofmind / vit-flax
Implementation of numerous Vision Transformers in Google's JAX and Flax.
☆20Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for vit-flax
- FID computation in Jax/Flax.☆24Updated 3 months ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated last year
- Little article showing how to load pytorch's models with linear memory consumption☆34Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 years ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆46Updated 3 months ago
- AdaCat☆49Updated 2 years ago
- ☆33Updated last year
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆46Updated 9 months ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 2 years ago
- Contrastive Language-Image Pretraining☆143Updated 2 years ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆43Updated last month
- Implementation of Vision Transformers in Flax☆17Updated 4 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆36Updated last month
- An open source implementation of CLIP.☆32Updated 2 years ago
- Local Attention - Flax module for Jax☆20Updated 3 years ago
- PyTorch interface for TrueGrad Optimizers☆39Updated last year
- A simple library for scaling up JAX programs☆125Updated last week
- Train vision models using JAX and 🤗 transformers☆95Updated 3 weeks ago
- ☆26Updated last year
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆86Updated last year
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆52Updated last year
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆142Updated 2 years ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- A metrics library for the JAX ecosystem☆39Updated last year
- This is a port of Mistral-7B model in JAX☆30Updated 4 months ago