Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
â267Jun 10, 2026Updated last week
Alternatives and similar repositories for rejax
Users that are interested in rejax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RLâ410Mar 18, 2026Updated 3 months ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ278Sep 22, 2025Updated 8 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ242Nov 24, 2025Updated 6 months ago
- Accelerated minigrid environments with JAXâ170Oct 20, 2025Updated 7 months ago
- â98Jan 21, 2026Updated 4 months ago
- Managed hosting for WordPress and PHP on Cloudways âĸ AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Clean single-file implementation of offline RL algorithms in JAXâ180Jun 5, 2026Updated last week
- RL Environments in JAX đâ901Apr 2, 2026Updated 2 months ago
- Drop-in environment replacements that make your RL algorithm train faster.â22Jun 19, 2024Updated last year
- Really Fast End-to-End Jax RL Implementationsâ1,081Sep 9, 2024Updated last year
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â341Dec 16, 2025Updated 6 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â412Feb 10, 2026Updated 4 months ago
- Baselines for gymnax đ¤â77Apr 3, 2023Updated 3 years ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.â258May 21, 2026Updated 3 weeks ago
- A framework for Reinforcement Learning research.â260Updated this week
- Managed Database hosting by DigitalOcean âĸ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Efficient baselines for autocurricula in JAX.â214Aug 24, 2024Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.â270Jun 6, 2026Updated last week
- Evolution Strategies in JAX đĻâ767Apr 2, 2026Updated 2 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithmsâ220Dec 19, 2025Updated 5 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ598May 11, 2026Updated last month
- An Open-Ended Agentic Simulatorâ60Aug 11, 2024Updated last year
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ840Jun 3, 2026Updated 2 weeks ago
- â95Feb 16, 2026Updated 4 months ago
- âī¸ Vectorized RL game environments in JAXâ617Mar 6, 2025Updated last year
- GPU virtual machines on DigitalOcean Gradient AI âĸ AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Benchmarking RL generalization in an interpretable way.â182Nov 20, 2025Updated 6 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learningâ80Aug 18, 2024Updated last year
- Multi-Agent Reinforcement Learning with JAXâ816Jun 10, 2026Updated last week
- Highly scalable 2D JAX physics engine.â68Apr 20, 2026Updated last month
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"â107Jan 23, 2025Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]â115Dec 5, 2023Updated 2 years ago
- POPGym Library in JAXâ14Apr 15, 2024Updated 2 years ago
- Accelerated replay buffers in JAXâ46Sep 17, 2022Updated 3 years ago
- Simple JAX Graphics Library.â38Nov 3, 2024Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI âĸ AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Modular Single-file Reinfocement Learning Algorithms Libraryâ38May 16, 2023Updated 3 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â756Oct 26, 2022Updated 3 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 3 years ago
- JAX implementation of RL algorithms and vectorized environmentsâ50Dec 26, 2023Updated 2 years ago
- Corax: Core RL in JAXâ41Feb 22, 2024Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.â46Updated this week
- â332Dec 19, 2024Updated last year