tech-srl / RASP-exps
Code for running the transformers in the ICML 2021 paper "Thinking Like Transformers"
☆16Updated 3 years ago
Alternatives and similar repositories for RASP-exps:
Users that are interested in RASP-exps are comparing it to the libraries listed below
- A programming language for formal/informal computation.☆41Updated this week
- One stop shop for all things carp☆59Updated 2 years ago
- ☆59Updated 3 years ago
- Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"☆60Updated 2 years ago
- Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.☆46Updated 3 years ago
- Mechanistic Interpretability for Transformer Models☆49Updated 2 years ago
- ☆59Updated 2 years ago
- Latent Diffusion Language Models☆68Updated last year
- An interpreter for RASP as described in the ICML 2021 paper "Thinking Like Transformers"☆301Updated 5 months ago
- The history files when recording human interaction while solving ARC tasks☆97Updated this week
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- Language-annotated Abstraction and Reasoning Corpus☆82Updated last year
- The Abstraction and Reasoning Corpus made into a web game☆89Updated 5 months ago
- Automatically take good care of your preemptible TPUs☆36Updated last year
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆237Updated last year
- A dataset of alignment research and code to reproduce it☆73Updated last year
- ☆16Updated 5 months ago
- A simple hypernetwork implementation in jax using haiku.☆23Updated 2 years ago
- Neural Networks for JAX☆83Updated 4 months ago
- Python library which enables complex compositions of language models such as scratchpads, chain of thought, tool use, selection-inference…☆202Updated last month
- Materials for ConceptARC paper☆88Updated 3 months ago
- ☆37Updated 2 years ago
- Proof-of-concept of global switching between numpy/jax/pytorch in a library.☆18Updated 8 months ago
- Resources from the EleutherAI Math Reading Group☆52Updated 2 months ago
- ☆19Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆44Updated 8 months ago
- Cellular Automata Accelerated in JAX (Oral at ICLR 2025).☆81Updated 3 months ago
- See the issue board for the current status of active and prospective projects!☆65Updated 3 years ago
- A JAX implementation of stochastic addition.☆12Updated 2 years ago
- A functional training loops library for JAX☆86Updated last year