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)
  ☆107Updated last month
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆58Updated last year
• FLAIROx / cultural-accumulation
  ☆16Updated last year
• nacloos / baba-is-ai
  Code for "Baba Is AI: Break the Rules to Beat the Benchmark"
  ☆41Updated 3 months ago
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆81Updated 3 years ago
• AllanYangZhou / universal_neural_functional
  ☆54Updated last year
• bolt-research / popgym-arcade
  Atari-style POMDPs
  ☆18Updated 2 weeks ago
• noanabeshima / tinymodel
  A TinyStories LM with SAEs and transcoders
  ☆14Updated 8 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆111Updated 2 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆54Updated 2 years ago
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆26Updated last year
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆72Updated 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
• nmonette / NCC-UED
  Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025
  ☆17Updated last month
• keyonvafa / world-model-evaluation
  ☆75Updated last year
• understanding-search / maze-dataset
  maze datasets for investigating OOD behavior of ML systems
  ☆67Updated 2 months ago
• danijar / diamond_env
  Standardized Minecraft Diamond Environment for Reinforcement Learning
  ☆32Updated 2 years ago
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆72Updated last year
• JacobPfau / procgenAISC
  ☆19Updated 2 years ago
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• ml-jku / helm
  ☆57Updated last year
• understanding-search / structured-representations-maze-transformers
  see github.com/understanding-search/maze-transformer
  ☆10Updated 2 years ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated last month
• amacrutherford / sampling-for-learnability
  Official codebase for "Sampling For Learnability", published at NeurIPS 2024
  ☆20Updated 2 months ago
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆44Updated last year
• Silent-Zebra / twisted-smc-lm
  ☆31Updated 9 months ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years ago
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆40Updated last year
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated 3 months ago