tech-srl / RASP
An interpreter for RASP as described in the ICML 2021 paper "Thinking Like Transformers"
☆307Updated 7 months ago
Alternatives and similar repositories for RASP:
Users that are interested in RASP are comparing it to the libraries listed below
- An interactive exploration of Transformer programming.☆262Updated last year
- ☆530Updated last year
- Neural Networks and the Chomsky Hierarchy☆205Updated last year
- ☆166Updated last year
- Python library which enables complex compositions of language models such as scratchpads, chain of thought, tool use, selection-inference…☆207Updated 3 months ago
- Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate …☆632Updated last year
- Emergent world representations: Exploring a sequence model trained on a synthetic task☆179Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆238Updated last year
- git extension for {collaborative, communal, continual} model development☆210Updated 5 months ago
- Named tensors with first-class dimensions for PyTorch☆320Updated last year
- See the issue board for the current status of active and prospective projects!☆65Updated 3 years ago
- Puzzles for exploring transformers☆342Updated last year
- Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.☆52Updated 3 years ago
- Erasing concepts from neural representations with provable guarantees☆227Updated 2 months ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆566Updated this week
- Mechanistic Interpretability Visualizations using React☆241Updated 4 months ago
- Language Modeling with the H3 State Space Model☆519Updated last year
- Train very large language models in Jax.☆203Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- seqax = sequence modeling + JAX☆153Updated last week
- Notebooks accompanying Anthropic's "Toy Models of Superposition" paper☆119Updated 2 years ago
- [NeurIPS 2023] Learning Transformer Programs☆159Updated 10 months ago
- Tools for understanding how transformer predictions are built layer-by-layer☆485Updated 10 months ago
- Understand and test language model architectures on synthetic tasks.☆191Updated last month
- Materials for ConceptARC paper☆91Updated 5 months ago
- ☆62Updated 2 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)☆187Updated 2 years ago
- Transformer Grammars: Augmenting Transformer Language Models with Syntactic Inductive Biases at Scale, TACL (2022)☆125Updated 5 months ago
- Language-annotated Abstraction and Reasoning Corpus☆86Updated last year
- JAX implementation of the Llama 2 model☆218Updated last year