gregorbachmann / scaling_mlps
☆51Updated 8 months ago
Alternatives and similar repositories for scaling_mlps:
Users that are interested in scaling_mlps are comparing it to the libraries listed below
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆44Updated last year
- Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]☆31Updated 5 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 3 years ago
- Code for the paper "On the Expressivity Role of LayerNorm in Transformers' Attention" (Findings of ACL'2023)☆45Updated 4 months ago
- Official code for the paper "Attention as a Hypernetwork"☆23Updated 7 months ago
- Latest Weight Averaging (NeurIPS HITY 2022)☆28Updated last year
- The official repository for our paper "The Dual Form of Neural Networks Revisited: Connecting Test Time Predictions to Training Patterns …☆16Updated last year
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆87Updated last year
- ☆51Updated 4 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 3 weeks ago
- ☆29Updated 2 years ago
- ☆33Updated 5 months ago
- This is a PyTorch implementation of the paperViP A Differentially Private Foundation Model for Computer Vision☆36Updated last year
- Fast training of unitary deep network layers from low-rank updates☆28Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- ☆20Updated last month
- ☆27Updated 7 months ago
- 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
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆79Updated last year
- Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]☆60Updated 4 months ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]☆18Updated last month
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆97Updated last year
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆47Updated 2 months ago
- FID computation in Jax/Flax.☆26Updated 6 months ago
- ☆41Updated last year
- Explorations into the recently proposed Taylor Series Linear Attention☆92Updated 5 months ago
- ☆54Updated 6 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- ☆30Updated 2 months ago