Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselines.
☆68Oct 30, 2026Updated this week
Alternatives and similar repositories for jax-baseline
Users that are interested in jax-baseline are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- JAxtar is a project with a JAX-native implementation of parallelizeable A* & Q* solver for neural heuristic search research.☆52Updated this week
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 10 months ago
- ☆130Feb 25, 2025Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆96Jun 4, 2024Updated 2 years ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆246Nov 24, 2025Updated 11 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.
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆425Mar 18, 2026Updated 7 months ago
- Clean single-file implementation of offline RL algorithms in JAX☆184Jun 5, 2026Updated 5 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆282Aug 27, 2026Updated 2 months ago
- Xtructure is datastructure for using in JAX☆24Updated this week
- POPGym Library in JAX☆14Apr 15, 2024Updated 2 years ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆113Nov 4, 2025Updated last year
- JAX implementation of NeuroEvolution of Augmenting Topologies (Neat)☆11Jul 6, 2024Updated 2 years ago
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆63Aug 3, 2023Updated 3 years ago
- GCRL in JAX. Official repository for LEO (ICML 2026).☆35Jun 20, 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.
- DR.Q is a debiased model-based representation reinforcement learning algorithm☆17Jun 25, 2026Updated 4 months ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- stable-baselines with JAX & Haiku☆13Jun 20, 2024Updated 2 years ago
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Feb 13, 2020Updated 6 years ago
- Unified Implementations of Offline Reinforcement Learning Algorithms☆236Dec 19, 2025Updated 10 months ago
- Source files to replicate experiments in my ICLR 2022 paper.☆76Jul 17, 2025Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆615Oct 27, 2026Updated last week
- Reinforcement learning in pure JAX.☆15Jun 24, 2026Updated 4 months ago
- Accelerated minigrid environments with JAX☆184Sep 10, 2026Updated last month
- 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.
- Skeleton for scalable and flexible Jax RL implementations☆102Jul 1, 2023Updated 3 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 3 years ago
- Prioritized Experience Replay implementation with proportional prioritization☆92Jul 18, 2023Updated 3 years ago
- Really Fast End-to-End Jax RL Implementations☆1,122Sep 9, 2024Updated 2 years ago
- Implemenation of the HIERarchical imagionation On Structured State Space Sequence Models (HIEROS) paper☆23Jul 14, 2024Updated 2 years ago
- Partially Observable Process Gym☆232Jun 11, 2026Updated 4 months ago
- Soft Actor-Critic implementation with SOTA model-free extension (REDQ) and SOTA model-based extension (MBPO).☆16Feb 21, 2021Updated 5 years ago
- HumEnv is an SMPL humanoid environment enabling systematic model comparison and reproducibility☆123Apr 22, 2025Updated last year
- ☆98Feb 16, 2026Updated 8 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Reinforcement Learning inside a 3D soccer simulation☆35Sep 15, 2024Updated 2 years ago
- Quadruped Robot controller design and simulation on Webots☆12Apr 28, 2020Updated 6 years ago
- Temporally Correlated Episodic Reinforcement Learning, ICLR 24☆12Apr 8, 2024Updated 2 years ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆117Oct 27, 2024Updated 2 years ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆26Aug 19, 2025Updated last year
- RL Environments in JAX 🌍☆940Oct 30, 2026Updated last week
- ☆14Updated this week