Alternatives and similar repositories for maze-transformer

Users that are interested in maze-transformer are comparing it to the libraries listed below

• jbloomAus / DecisionTransformerInterpretability
  Interpreting how transformers simulate agents performing RL tasks
  ☆89Updated 2 years ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆106Updated 2 weeks ago
• noanabeshima / tinymodel
  A TinyStories LM with SAEs and transcoders
  ☆14Updated 8 months ago
• amacrutherford / sampling-for-learnability
  Official codebase for "Sampling For Learnability", published at NeurIPS 2024
  ☆20Updated last month
• nacloos / baba-is-ai
  Code for "Baba Is AI: Break the Rules to Beat the Benchmark"
  ☆41Updated 3 months ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆71Updated 11 months ago
• understanding-search / maze-dataset
  maze datasets for investigating OOD behavior of ML systems
  ☆66Updated last month
• FLAIROx / cultural-accumulation
  ☆15Updated last year
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆56Updated last year
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆29Updated last year
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆25Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆80Updated 3 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆54Updated 2 years ago
• AllanYangZhou / universal_neural_functional
  ☆53Updated last year
• understanding-search / structured-representations-maze-transformers
  see github.com/understanding-search/maze-transformer
  ☆10Updated 2 years ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated 2 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆110Updated 2 years ago
• JacobPfau / procgenAISC
  ☆19Updated 2 years ago
• nmonette / NCC-UED
  Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025
  ☆17Updated 2 weeks ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• callummcdougall / sae-exercises-mats
  ☆24Updated last year
• ucl-dark / pax
  Scalable Opponent Shaping Experiments in JAX
  ☆24Updated last year
• maxencefaldor / learned-qd
  Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization
  ☆20Updated last week
• timaeus-research / devinterp
  Tools for studying developmental interpretability in neural networks.
  ☆117Updated 5 months ago
• taufeeque9 / codebook-features
  Sparse and discrete interpretability tool for neural networks
  ☆64Updated last year
• adamkarvonen / SAE_BoardGameEval
  ☆23Updated 10 months ago
• noanabeshima / matryoshka-saes
  ☆25Updated last year
• likenneth / othello_world
  Emergent world representations: Exploring a sequence model trained on a synthetic task
  ☆196Updated 2 years ago
• astanic / crafter-ood
  ☆19Updated 3 years ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years ago