mcleish7 / arithmeticLinks
Code to reproduce "Transformers Can Do Arithmetic with the Right Embeddings", McLeish et al (NeurIPS 2024)
☆198Updated last year
Alternatives and similar repositories for arithmetic
Users that are interested in arithmetic are comparing it to the libraries listed below
Sorting:
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆186Updated 2 weeks ago
- Understand and test language model architectures on synthetic tasks.☆252Updated 3 weeks ago
- ☆152Updated 4 months ago
- Code for NeurIPS'24 paper 'Grokked Transformers are Implicit Reasoners: A Mechanistic Journey to the Edge of Generalization'☆235Updated 6 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆247Updated 8 months ago
- A MAD laboratory to improve AI architecture designs 🧪☆137Updated last year
- ☆207Updated 3 weeks ago
- Repository for the paper Stream of Search: Learning to Search in Language☆152Updated last year
- Functional Benchmarks and the Reasoning Gap☆89Updated last year
- nanoGPT-like codebase for LLM training☆113Updated 3 months ago
- OpenCoconut implements a latent reasoning paradigm where we generate thoughts before decoding.☆175Updated last year
- Extract full next-token probabilities via language model APIs☆248Updated last year
- $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources☆150Updated 4 months ago
- PyTorch library for Active Fine-Tuning☆96Updated 4 months ago
- Applying SAEs for fine-grained control☆25Updated last year
- A curated reading list of research in Adaptive Computation, Inference-Time Computation & Mixture of Experts (MoE).☆160Updated last year
- RuLES: a benchmark for evaluating rule-following in language models☆248Updated 11 months ago
- Open source interpretability artefacts for R1.☆170Updated 9 months ago
- ☆91Updated last year
- ModuleFormer is a MoE-based architecture that includes two different types of experts: stick-breaking attention heads and feedforward exp…☆226Updated 4 months ago
- Language models scale reliably with over-training and on downstream tasks☆99Updated last year
- Experiments for efforts to train a new and improved t5☆76Updated last year
- ☆74Updated last year
- Implementation of the Llama architecture with RLHF + Q-learning☆170Updated last year
- Universal Neurons in GPT2 Language Models☆30Updated last year
- Code for reproducing our paper "Not All Language Model Features Are Linear"☆83Updated last year
- ☆56Updated last year
- ☆185Updated 2 years ago
- Replicating O1 inference-time scaling laws☆92Updated last year