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
• AllanYangZhou / universal_neural_functional
  ☆52Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆79Updated 3 years ago
• understanding-search / structured-representations-maze-transformers
  see github.com/understanding-search/maze-transformer
  ☆10Updated last year
• clement-bonnet / lpn
  Latent Program Network (from the "Searching Latent Program Spaces" paper)
  ☆93Updated 5 months ago
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆52Updated last year
• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆47Updated 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
  ☆16Updated 2 months ago
• understanding-search / maze-dataset
  maze datasets for investigating OOD behavior of ML systems
  ☆53Updated last week
• 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 11 months ago
• kvfrans / powderworld
  Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions
  ☆68Updated 11 months ago
• noanabeshima / tinymodel
  A TinyStories LM with SAEs and transcoders
  ☆13Updated 4 months ago
• luchris429 / discovered-policy-optimisation
  Code for Discovered Policy Optimisation (NeurIPS 2022)
  ☆11Updated 2 years ago
• google-deepmind / pushworld
  PushWorld: A benchmark for manipulation planning with tools and movable obstacles
  ☆83Updated last year
• quasimetric-learning / torch-quasimetric
  PyTorch Package For Quasimetric Learning
  ☆42Updated 9 months ago
• FLAIROx / cultural-accumulation
  ☆13Updated last year
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago
• danijar / elements
  Building blocks for productive research
  ☆59Updated 3 weeks ago
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated last year
• FLAIROx / jaxirl
  Contains JAX implementation of algorithms for inverse reinforcement learning
  ☆73Updated last year
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• maxencefaldor / omni-epic
  OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).
  ☆67Updated 8 months ago
• FLAIROx / JaxGL
  Simple JAX Graphics Library.
  ☆36Updated 9 months ago
• danijar / ninjax
  General Modules for JAX
  ☆67Updated 4 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated last month
• HumanCompatibleAI / leela-interp
  Code for "Evidence of Learned Look-Ahead in a Chess-Playing Neural Network"
  ☆25Updated last year
• ml-jku / helm
  ☆55Updated 9 months ago
• DramaCow / jaxued
  ☆82Updated 5 months ago