JAX implementations of core Deep RL algorithms
☆84May 2, 2022Updated 3 years ago
Alternatives and similar repositories for jax-rl
Users that are interested in jax-rl are comparing it to the libraries listed below
Sorting:
- A collection of RL algorithms written in JAX.☆105Jul 5, 2022Updated 3 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆753Oct 26, 2022Updated 3 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆59Aug 4, 2022Updated 3 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- Modular framework for Reinforcement Learning in python☆183Feb 1, 2023Updated 3 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- A JAX Implementation of the Twin Delayed DDPG Algorithm☆35Mar 12, 2020Updated 5 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆105May 17, 2022Updated 3 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- RL Environments in JAX 🌍☆868May 30, 2025Updated 9 months ago
- GPT implementation in Flax☆18Jan 8, 2022Updated 4 years ago
- A modular implementation of PPO, and soon hopefully other algorithms.☆26Jan 16, 2024Updated 2 years ago
- ☆1,409Updated this week
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆26Feb 2, 2025Updated last year
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- ☆52Jan 20, 2023Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆263Oct 31, 2025Updated 4 months ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- ☆63Jan 30, 2026Updated last month
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆273Sep 22, 2025Updated 5 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- ☆134May 8, 2020Updated 5 years ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,274Aug 12, 2024Updated last year
- Behavioural cloning solution to MineRL2020 competition☆18Mar 6, 2021Updated 5 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- Accelerated minigrid environments with JAX☆162Oct 20, 2025Updated 4 months ago
- PyTorch implementation of Soft Actor-Critic + Autoencoder(SAC+AE)☆254May 3, 2020Updated 5 years ago
- On the model-based stochastic value gradient for continuous reinforcement learning☆57Updated this week
- Episodic Control☆22Sep 20, 2022Updated 3 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- Learning Action-Value Gradients in Model-based Policy Optimization☆32Sep 7, 2021Updated 4 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago