Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
â274Jun 10, 2026Updated last month
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â416Mar 18, 2026Updated 4 months ago
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ279Sep 22, 2025Updated 10 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ242Nov 24, 2025Updated 8 months ago
- Accelerated minigrid environments with JAXâ175Oct 20, 2025Updated 9 months ago
- RL Environments in JAX đâ910Apr 2, 2026Updated 3 months ago
- 1-Click AI Models by DigitalOcean Gradient âĸ AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- â98Jan 21, 2026Updated 6 months ago
- Clean single-file implementation of offline RL algorithms in JAXâ182Jun 5, 2026Updated last month
- Drop-in environment replacements that make your RL algorithm train faster.â22Jun 19, 2024Updated 2 years ago
- Really Fast End-to-End Jax RL Implementationsâ1,093Sep 9, 2024Updated last year
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â342Dec 16, 2025Updated 7 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â431Jun 20, 2026Updated last month
- Baselines for gymnax đ¤â78Apr 3, 2023Updated 3 years ago
- Efficient baselines for autocurricula in JAX.â214Aug 24, 2024Updated last year
- Unified Implementations of Offline Reinforcement Learning Algorithmsâ225Dec 19, 2025Updated 7 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer âĸ AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Evolution Strategies in JAX đĻâ778Apr 2, 2026Updated 3 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"â107Jan 23, 2025Updated last year
- A framework for Reinforcement Learning research.â268Updated this week
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.â274Jun 6, 2026Updated last month
- A jax-native environment interfaceâ17Jul 21, 2026Updated last week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ603Updated this week
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ854Jun 18, 2026Updated last month
- An Open-Ended Agentic Simulatorâ61Aug 11, 2024Updated last year
- â96Feb 16, 2026Updated 5 months ago
- Managed Database hosting by DigitalOcean âĸ AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- âī¸ Vectorized RL game environments in JAXâ634Mar 6, 2025Updated last year
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.â262May 21, 2026Updated 2 months ago
- Multi-Agent Reinforcement Learning with JAXâ837Jul 21, 2026Updated last week
- Benchmarking RL generalization in an interpretable way.â183Nov 20, 2025Updated 8 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learningâ81Aug 18, 2024Updated last year
- Highly scalable 2D JAX physics engine.â68Apr 20, 2026Updated 3 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]â117Dec 5, 2023Updated 2 years ago
- Partially Observable Benchmarks in JAXâ25Apr 30, 2026Updated 2 months ago
- POPGym Library in JAXâ14Apr 15, 2024Updated 2 years ago
- AI Agents on DigitalOcean Gradient AI Platform âĸ AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Accelerated replay buffers in JAXâ46Sep 17, 2022Updated 3 years ago
- Modular Single-file Reinfocement Learning Algorithms Libraryâ38May 16, 2023Updated 3 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 3 years ago
- Simple JAX Graphics Library.â38Nov 3, 2024Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â757Oct 26, 2022Updated 3 years ago
- GCRL in JAX. Official repository for LEO (ICML 2026).â28Jun 20, 2026Updated last month
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.â47Jul 10, 2026Updated 2 weeks ago