srush / raspyLinks
An interactive exploration of Transformer programming.
☆271Updated 2 years ago
Alternatives and similar repositories for raspy
Users that are interested in raspy are comparing it to the libraries listed below
Sorting:
- An interpreter for RASP as described in the ICML 2021 paper "Thinking Like Transformers"☆323Updated last year
- Puzzles for exploring transformers☆385Updated 2 years ago
- git extension for {collaborative, communal, continual} model development☆217Updated last year
- ☆551Updated last year
- ☆490Updated last year
- A puzzle to learn about prompting☆135Updated 2 years ago
- Extract full next-token probabilities via language model APIs☆248Updated last year
- ☆289Updated last year
- Resources from the EleutherAI Math Reading Group☆54Updated 11 months ago
- Functional local implementations of main model parallelism approaches☆95Updated 2 years ago
- Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.☆59Updated 2 months ago
- Minimalistic, extremely fast, and hackable researcher's toolbench for GPT models in 307 lines of code. Reaches <3.8 validation loss on wi…☆355Updated last year
- seqax = sequence modeling + JAX☆170Updated 6 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆694Updated this week
- Train very large language models in Jax.☆210Updated 2 years ago
- Python library which enables complex compositions of language models such as scratchpads, chain of thought, tool use, selection-inference…☆216Updated 2 weeks ago
- Automatic gradient descent☆217Updated 2 years ago
- Fast bare-bones BPE for modern tokenizer training☆174Updated 7 months ago
- 🧱 Modula software package☆322Updated 5 months ago
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆181Updated 8 months ago
- Simple Transformer in Jax☆142Updated last year
- JAX implementation of the Llama 2 model☆216Updated last year
- Neural Networks and the Chomsky Hierarchy☆212Updated last year
- ☆167Updated 2 years ago
- Notebooks accompanying Anthropic's "Toy Models of Superposition" paper☆133Updated 3 years ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆298Updated last year
- ☆22Updated last year
- ☆152Updated 4 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- Inference code for LLaMA models in JAX☆120Updated last year