mcleish7 / arithmetic
Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
☆178Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for arithmetic
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆84Updated last week
- Understand and test language model architectures on synthetic tasks.☆162Updated 6 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆95Updated 6 months ago
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆95Updated 2 weeks ago
- Create feature-centric and prompt-centric visualizations for sparse autoencoders (like those from Anthropic's published research).☆157Updated last month
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆113Updated 7 months ago
- ☆107Updated this week
- Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead☆109Updated last week
- ☆101Updated 3 months ago
- Tree Attention: Topology-aware Decoding for Long-Context Attention on GPU clusters☆104Updated last month
- Language models scale reliably with over-training and on downstream tasks☆94Updated 7 months ago
- code for training & evaluating Contextual Document Embedding models☆117Updated this week
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆214Updated 3 months ago
- Code for reproducing our paper "Not All Language Model Features Are Linear"☆61Updated last week
- RuLES: a benchmark for evaluating rule-following in language models☆211Updated last month
- ☆175Updated this week
- ☆53Updated 10 months ago
- Extract full next-token probabilities via language model APIs☆229Updated 8 months ago
- Code for NeurIPS'24 paper 'Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalization'☆161Updated last month
- Implementation of the Llama architecture with RLHF + Q-learning☆157Updated 10 months ago
- Scalable neural net training via automatic normalization in the modular norm.☆121Updated 3 months ago
- Normalized Transformer (nGPT)☆66Updated this week
- WIP☆89Updated 3 months ago
- ☆105Updated last month
- ☆161Updated last year
- ☆102Updated last month
- Implementation of CALM from the paper "LLM Augmented LLMs: Expanding Capabilities through Composition", out of Google Deepmind☆169Updated 2 months ago
- Code repository for the c-BTM paper☆105Updated last year
- Erasing concepts from neural representations with provable guarantees☆209Updated last week
- Experiments for efforts to train a new and improved t5☆76Updated 7 months ago