FLAIROx / Kinetix
Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.
☆129Updated last week
Alternatives and similar repositories for Kinetix:
Users that are interested in Kinetix are comparing it to the libraries listed below
- Highly scalable 2D JAX physics engine.☆45Updated 3 weeks ago
- ☆71Updated 6 months ago
- ☆188Updated 2 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆137Updated 2 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆121Updated this week
- Clean single-file implementation of offline RL algorithms in JAX☆132Updated last month
- Contains JAX implementation of algorithms for inverse reinforcement learning☆69Updated 6 months ago
- Efficient baselines for autocurricula in JAX.☆179Updated 5 months ago
- ☆41Updated 7 months ago
- ☆73Updated 3 months ago
- Baselines for gymnax 🤖☆63Updated last year
- Accelerated minigrid environments with JAX☆128Updated 6 months ago
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- Simple JAX Graphics Library.☆30Updated 3 months ago
- Jax/Flax Implementation of TD-MPC2☆55Updated this week
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆225Updated last month
- A framework for Reinforcement Learning research.☆136Updated 2 weeks ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆277Updated this week
- Foundation Policies with Hilbert Representations (ICML 2024)☆76Updated 10 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆64Updated 8 months ago
- ☆48Updated 2 years ago
- General Modules for JAX☆63Updated 6 months ago
- Skeleton for scalable and flexible Jax RL implementations☆72Updated last year
- JAX implementation of RL algorithms and vectorized environments☆39Updated last year
- ☆70Updated last year
- ☆70Updated 4 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 5 months ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆111Updated this week
- A tool for recording RL trajectories.☆101Updated 3 months ago
- ☆259Updated 2 years ago