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
Related projects: ⓘ
- An open source implementation of CLIP.☆32Updated last year
- FID computation in Jax/Flax.☆23Updated 2 months ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated last year
- ☆40Updated this week
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- ☆38Updated this week
- ☆73Updated last year
- Hacks for PyTorch☆17Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆45Updated last month
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆74Updated 2 years ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…☆115Updated last month
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆56Updated 10 months ago
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- ☆29Updated last year
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆47Updated 2 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 2 years ago
- Some personal experiments around routing tokens to different autoregressive attention, akin to mixture-of-experts☆101Updated last year
- Load any clip model with a standardized interface☆21Updated 4 months ago
- PyTorch interface for TrueGrad Optimizers☆39Updated last year
- JAX implementation ViT-VQGAN☆77Updated last year
- A dashboard for exploring timm learning rate schedulers☆18Updated last year
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 2 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 2 years ago
- Various transformers for FSDP research☆31Updated last year
- Utilities for PyTorch distributed☆23Updated 11 months ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆40Updated 7 months ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆43Updated last year