understanding-search / maze-transformer
This repo is built to facilitate the training and analysis of autoregressive transformers on maze-solving tasks.
☆26Updated 6 months ago
Alternatives and similar repositories for maze-transformer:
Users that are interested in maze-transformer are comparing it to the libraries listed below
- maze datasets for investigating OOD behavior of ML systems☆31Updated this week
- Interpreting how transformers simulate agents performing RL tasks☆77Updated last year
- ☆13Updated 7 months ago
- Scaling scaling laws with board games.☆48Updated last year
- unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"☆75Updated 2 years ago
- Investigating the generalization behavior of LM probes trained to predict truth labels: (1) from one annotator to another, and (2) from e…☆26Updated 9 months ago
- PyTorch and NNsight implementation of AtP* (Kramar et al 2024, DeepMind)☆18Updated last month
- Sparse Autoencoder Training Library☆42Updated 4 months ago
- Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"☆20Updated 9 months ago
- ☆26Updated last year
- Open source replication of Anthropic's Crosscoders for Model Diffing☆44Updated 4 months ago
- A library for efficient patching and automatic circuit discovery.☆56Updated 3 weeks ago
- ☆49Updated last year
- An Open-Ended Agentic Simulator☆43Updated 7 months ago
- Tools for studying developmental interpretability in neural networks.☆86Updated last month
- Universal Neurons in GPT2 Language Models☆27Updated 9 months ago
- ☆61Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆9Updated last year
- Code for minimum-entropy coupling.☆31Updated 8 months ago
- ☆26Updated 11 months ago
- ☆21Updated last month
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago
- [ICML 2024] Official code release accompanying the paper "diff History for Neural Language Agents" (Piterbarg, Pinto, Fergus)☆20Updated 6 months ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 4 months ago
- The Energy Transformer block, in JAX☆56Updated last year
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Updated 7 months ago
- we got you bro☆35Updated 7 months ago
- A collection of meta-learning algorithms in Jax☆22Updated 2 years ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆67Updated 3 months ago
- ☆60Updated 3 months ago