Leiay / looped_transformerLinks
☆32Updated last year
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"☆84Updated 11 months ago
- ☆53Updated last year
- nanoGPT-like codebase for LLM training☆107Updated 4 months ago
- ☆85Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆164Updated 3 months ago
- Stick-breaking attention☆60Updated 3 months ago
- ☆83Updated last year
- Official repository for our paper, Transformers Learn Higher-Order Optimization Methods for In-Context Learning: A Study with Linear Mode…☆18Updated 10 months ago
- ☆40Updated last month
- Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)☆81Updated 2 years ago
- Muon fsdp 2☆44Updated last month
- ☆50Updated last year
- Mamba support for transformer lens☆18Updated last year
- NeurIPS 2024 tutorial on LLM Inference☆47Updated 9 months ago
- Official implementation of the transformer (TF) architecture suggested in a paper entitled "Looped Transformers as Programmable Computers…☆27Updated 2 years ago
- Code and Configs for Asynchronous RLHF: Faster and More Efficient RL for Language Models☆63Updated 5 months ago
- AdaSplash: Adaptive Sparse Flash Attention (aka Flash Entmax Attention)☆26Updated 2 months ago
- The evaluation framework for training-free sparse attention in LLMs☆98Updated 3 months ago
- Official Pytorch Implementation of "The Curse of Depth in Large Language Models" by Wenfang Sun, Xinyuan Song, Pengxiang Li, Lu Yin,Yefen…☆57Updated 2 months ago
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining (NeurIPS 2024)