Leiay / looped_transformerLinks
☆33Updated 2 years ago
Alternatives and similar repositories for looped_transformer
Users that are interested in looped_transformer are comparing it to the libraries listed below
Sorting:
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆86Updated last year
- ☆53Updated last year
- ☆91Updated last year
- Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…☆20Updated last year
- Official implementation of the transformer (TF) architecture suggested in a paper entitled "Looped Transformers as Programmable Computers…☆29Updated 2 years ago
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆181Updated 6 months ago
- ☆50Updated last week
- Unofficial Implementation of Selective Attention Transformer☆18Updated last year
- ☆84Updated 2 years ago
- ☆20Updated last month
- Stick-breaking attention☆62Updated 5 months ago
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining (NeurIPS 2024)☆38Updated last year
- nanoGPT-like codebase for LLM training☆113Updated last month
- ☆35Updated last year
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆63Updated 9 months ago
- ☆107Updated last year
- ☆51Updated last year
- [NeurIPS 2024] Low rank memory efficient optimizer without SVD☆32Updated 5 months ago
- Language models scale reliably with over-training and on downstream tasks☆100Updated last year
- Official Pytorch Implementation of "The Curse of Depth in Large Language Models" by Wenfang Sun, Xinyuan Song, Pengxiang Li, Lu Yin,Yefen…☆64Updated 3 weeks ago
- ☆38Updated last year
- ☆33Updated last year
- ☆33Updated 11 months ago
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆27Updated last year
- Gemstones: A Model Suite for Multi-Faceted Scaling Laws (NeurIPS 2025)☆30Updated 2 months ago
- Some preliminary explorations of Mamba's context scaling.☆218Updated last year
- ☆23Updated 10 months ago
- [ICLR 2025] Code for the paper "Beyond Autoregression: Discrete Diffusion for Complex Reasoning and Planning"☆86Updated 10 months ago
- Parallelizing non-linear sequential models over the sequence length☆56Updated 6 months ago