Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
☆234Feb 26, 2026Updated 3 weeks ago
Alternatives and similar repositories for Kinetix
Users that are interested in Kinetix are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Highly scalable 2D JAX physics engine.☆64Feb 20, 2026Updated last month
- Simple JAX Graphics Library.☆37Nov 3, 2024Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆237Nov 24, 2025Updated 3 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
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆264Oct 31, 2025Updated 4 months ago
- Scalable Opponent Shaping Experiments in JAX☆25Apr 13, 2024Updated last year
- ☆94Jan 21, 2026Updated 2 months ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆18Nov 24, 2025Updated 3 months ago
- An Open-Ended Agentic Simulator☆60Aug 11, 2024Updated last year
- Efficient baselines for autocurricula in JAX.☆209Aug 24, 2024Updated last year
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆381Feb 10, 2026Updated last month
- RL Environments in JAX 🌍☆871May 30, 2025Updated 9 months ago
- Accelerated minigrid environments with JAX☆163Oct 20, 2025Updated 5 months ago
- Really Fast End-to-End Jax RL Implementations☆1,035Sep 9, 2024Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆103Jan 23, 2025Updated last year
- Clean single-file implementation of offline RL algorithms in JAX☆174Nov 24, 2025Updated 3 months ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 5 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆274Sep 22, 2025Updated 6 months ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆326Dec 16, 2025Updated 3 months ago
- ☆23Aug 19, 2022Updated 3 years ago
- ☆16Jul 16, 2024Updated last year
- Code for Model-Free Opponent Shaping (ICML 2022)☆20Nov 18, 2022Updated 3 years ago
- ☆93Feb 16, 2026Updated last month
- Unified Implementations of Offline Reinforcement Learning Algorithms☆203Dec 19, 2025Updated 3 months ago
- ☆22May 12, 2025Updated 10 months ago
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆252Nov 23, 2025Updated 4 months ago
- Evolution Strategies in JAX 🦎☆739Sep 20, 2025Updated 6 months ago
- Multi-Agent Reinforcement Learning with JAX☆772Jan 14, 2026Updated 2 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆820Mar 9, 2026Updated 2 weeks ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- OMNI-EPIC: Open-endedness via Models of human Notions of Interestingness with Environments Programmed in Code (ICLR 2025).☆74Dec 26, 2024Updated last year
- The official implementation of "Horizon Reduction Makes RL Scalable"☆182Aug 2, 2025Updated 7 months ago
- Baselines for gymnax 🤖☆75Apr 3, 2023Updated 2 years ago
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆82Feb 13, 2025Updated last year
- A framework for Reinforcement Learning research.☆252Updated this week
- Contains JAX implementation of algorithms for inverse reinforcement learning☆74Aug 18, 2024Updated last year
- ♟️ Vectorized RL game environments in JAX☆593Mar 6, 2025Updated last year
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 2 years ago