(Crafter + NetHack) in JAX. ICML 2024 Spotlight.
â381Feb 10, 2026Updated last month
Alternatives and similar repositories for Craftax
Users that are interested in Craftax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- â94Jan 21, 2026Updated 2 months ago
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ397Mar 1, 2026Updated 3 weeks ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â326Dec 16, 2025Updated 3 months ago
- â22May 12, 2025Updated 10 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ238Nov 24, 2025Updated 4 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!â264Oct 31, 2025Updated 4 months ago
- Efficient baselines for autocurricula in JAX.â209Aug 24, 2024Updated last year
- RL Environments in JAX đâ871May 30, 2025Updated 9 months ago
- Accelerated minigrid environments with JAXâ163Oct 20, 2025Updated 5 months ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ274Sep 22, 2025Updated 6 months ago
- Really Fast End-to-End Jax RL Implementationsâ1,035Sep 9, 2024Updated last year
- â93Feb 16, 2026Updated last month
- âī¸ Vectorized RL game environments in JAXâ593Mar 6, 2025Updated last year
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.â234Feb 26, 2026Updated 3 weeks ago
- Simple JAX Graphics Library.â37Nov 3, 2024Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ820Mar 9, 2026Updated 2 weeks ago
- Multi-Agent Reinforcement Learning with JAXâ772Jan 14, 2026Updated 2 months ago
- Benchmarking the Spectrum of Agent Capabilitiesâ528Jan 23, 2024Updated 2 years ago
- Baselines for gymnax đ¤â75Apr 3, 2023Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithmsâ165Jun 23, 2023Updated 2 years ago
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.â252Nov 23, 2025Updated 4 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ122Aug 22, 2024Updated last year
- â122Feb 25, 2025Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]â112Dec 5, 2023Updated 2 years ago
- Accelerated replay buffers in JAXâ46Sep 17, 2022Updated 3 years ago
- An Open-Ended Agentic Simulatorâ60Aug 11, 2024Updated last year
- Clean single-file implementation of offline RL algorithms in JAXâ174Nov 24, 2025Updated 4 months ago
- đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXâ61Oct 23, 2023Updated 2 years ago
- Highly scalable 2D JAX physics engine.â64Feb 20, 2026Updated last month
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ571Feb 25, 2026Updated last month
- POPGym Library in JAXâ12Apr 15, 2024Updated last year
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"â94Nov 4, 2025Updated 4 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.â868Aug 12, 2024Updated last year
- A collection of matrix games in JAXâ13Nov 28, 2024Updated last year
- â10Jun 27, 2024Updated last year
- Evolution Strategies in JAX đĻâ739Sep 20, 2025Updated 6 months ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 2 years ago
- Partially Observable Process Gymâ213Jun 12, 2025Updated 9 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"â103Jan 23, 2025Updated last year