jysohn1108 / Looped-Transformer
Official implementation of the transformer (TF) architecture suggested in a paper entitled "Looped Transformers as Programmable Computers"
☆24Updated last year
Alternatives and similar repositories for Looped-Transformer:
Users that are interested in Looped-Transformer are comparing it to the libraries listed below
- ☆24Updated last year
- Official repository of paper "RNNs Are Not Transformers (Yet): The Key Bottleneck on In-context Retrieval"☆26Updated 11 months ago
- ☆81Updated last year
- Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"☆44Updated last month
- ☆51Updated 10 months ago
- ☆45Updated last year
- Universal Neurons in GPT2 Language Models☆27Updated 9 months ago
- Reference implementation for Reward-Augmented Decoding: Efficient Controlled Text Generation With a Unidirectional Reward Model☆42Updated last year
- Q-Probe: A Lightweight Approach to Reward Maximization for Language Models☆41Updated 9 months ago
- [NeurIPS 2024 Spotlight] Code and data for the paper "Finding Transformer Circuits with Edge Pruning".☆47Updated 2 weeks ago
- Code for reproducing our paper "Not All Language Model Features Are Linear"☆72Updated 4 months ago
- ☆30Updated 5 months ago
- ☆47Updated last year
- ☆60Updated 11 months ago
- Advantage Leftover Lunch Reinforcement Learning (A-LoL RL): Improving Language Models with Advantage-based Offline Policy Gradients☆26Updated 6 months ago
- Stick-breaking attention☆49Updated last week
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆71Updated 4 months ago
- Self-Supervised Alignment with Mutual Information☆16Updated 10 months ago
- ☆74Updated 7 months ago
- ☆30Updated 2 months ago
- ☆17Updated 5 months ago
- The repository contains code for Adaptive Data Optimization☆20Updated 3 months ago
- Code for the arXiv preprint "The Unreasonable Effectiveness of Easy Training Data"☆46Updated last year
- ☆18Updated 8 months ago
- ☆12Updated last year
- ☆21Updated last month
- ☆26Updated 8 months ago
- NeurIPS 2024 tutorial on LLM Inference☆39Updated 3 months ago
- ☆49Updated 7 months ago
- A Large-Scale, High-Quality Math Dataset for Reinforcement Learning in Language Models☆44Updated last month