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
  ☆87Updated last year
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆52Updated last year
• 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
• nacloos / baba-is-ai
  ☆39Updated 2 weeks ago
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆96Updated 6 months ago
• noanabeshima / tinymodel
  A TinyStories LM with SAEs and transcoders
  ☆13Updated 5 months ago
• nmonette / NCC-UED
  Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025
  ☆15Updated last month
• amacrutherford / sampling-for-learnability
  Official codebase for "Sampling For Learnability", published at NeurIPS 2024
  ☆18Updated 3 months ago
• bolt-research / popgym-arcade
  Atari-style POMDPs
  ☆15Updated last week
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆48Updated 2 years ago
• FLAIROx / cultural-accumulation
  ☆13Updated last year
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆68Updated last year
• understanding-search / structured-representations-maze-transformers
  see github.com/understanding-search/maze-transformer
  ☆10Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆79Updated 3 years ago
• ucl-dark / pax
  Scalable Opponent Shaping Experiments in JAX
  ☆24Updated last year
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆12Updated 2 years ago
• brownirl / lambda_discrepancy
  Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy
  ☆20Updated 10 months ago
• danijar / ninjax
  General Modules for JAX
  ☆67Updated last week
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 weeks ago
• facebookresearch / gen_dgrl
  Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024
  ☆28Updated last year
• maxencefaldor / learned-qd
  Discovering Quality-Diversity Algorithms via Meta-Black-Box Optimization
  ☆17Updated 5 months ago
• FLAIROx / JaxGL
  Simple JAX Graphics Library.
  ☆36Updated 10 months ago
• yilundu / ired_code_release
  ☆70Updated last year
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated last year
• danijar / elements
  Building blocks for productive research
  ☆60Updated last month
• ml-jku / helm
  ☆55Updated 10 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆59Updated last year
• timaeus-research / devinterp
  Tools for studying developmental interpretability in neural networks.
  ☆103Updated 2 months ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆21Updated last year