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
  ☆88Updated 2 years ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆105Updated last month
• nacloos / baba-is-ai
  ☆41Updated 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 last year
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆54Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆79Updated 3 years ago
• understanding-search / maze-dataset
  maze datasets for investigating OOD behavior of ML systems
  ☆64Updated last month
• AllanYangZhou / universal_neural_functional
  ☆53Updated last year
• FLAIROx / cultural-accumulation
  ☆15Updated last year
• understanding-search / structured-representations-maze-transformers
  see github.com/understanding-search/maze-transformer
  ☆10Updated last year
• noanabeshima / tinymodel
  A TinyStories LM with SAEs and transcoders
  ☆13Updated 7 months ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆28Updated last year
• bolt-research / popgym-arcade
  Atari-style POMDPs
  ☆17Updated last week
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆70Updated last year
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• FLAIROx / JaxGL
  Simple JAX Graphics Library.
  ☆36Updated last year
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years ago
• bilal-chughtai / rep-theory-mech-interp
  ☆27Updated 2 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆53Updated 2 years ago
• amacrutherford / sampling-for-learnability
  Official codebase for "Sampling For Learnability", published at NeurIPS 2024
  ☆20Updated last month
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆28Updated last year
• FLAIROx / popjym
  POPGym Library in JAX
  ☆11Updated last year
• nmonette / NCC-UED
  Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025
  ☆17Updated 3 months ago
• brownirl / lambda_discrepancy
  Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
  ☆21Updated last year
• ucl-dark / pax
  Scalable Opponent Shaping Experiments in JAX
  ☆24Updated last year
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated last month
• twitter-research / hyperbolic-rl
  ☆57Updated 3 years ago
• MichaelTMatthews / Jax2D
  Highly scalable 2D JAX physics engine.
  ☆60Updated 8 months ago
• ml-jku / helm
  ☆57Updated last year
• DramaCow / jaxued
  ☆87Updated 2 months ago