☆52Jan 20, 2023Updated 3 years ago
Alternatives and similar repositories for jaxrl2
Users that are interested in jaxrl2 are comparing it to the libraries listed below
Sorting:
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆30Oct 26, 2022Updated 3 years ago
- Reinforcement learning library in JAX.☆101Oct 22, 2023Updated 2 years ago
- Corax: Core RL in JAX☆40Feb 22, 2024Updated 2 years ago
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆105May 17, 2022Updated 3 years ago
- ☆23Aug 19, 2022Updated 3 years ago
- ☆80Dec 9, 2022Updated 3 years ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆113May 27, 2024Updated last year
- Clean single-file implementation of offline RL algorithms in JAX☆170Nov 24, 2025Updated 3 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆753Oct 26, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 3 months ago
- Conservative Q learning in Jax☆57Feb 7, 2023Updated 3 years ago
- JAX implementation of RL algorithms and vectorized environments☆51Dec 26, 2023Updated 2 years ago
- GPT implementation in Flax☆18Jan 8, 2022Updated 4 years ago
- ☆20Mar 19, 2024Updated last year
- ☆18May 30, 2023Updated 2 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- A simple and easy to use implementation of the soft actor-critic algorithm.☆15Sep 2, 2022Updated 3 years ago
- ☆24Jan 26, 2024Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.☆21Sep 18, 2020Updated 5 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Author's implementation of ReBRAC, a minimalist improvement upon TD3+BC☆62Aug 3, 2023Updated 2 years ago
- Change-Based Exploration Transfer☆35Apr 24, 2022Updated 3 years ago
- ☆35Nov 22, 2024Updated last year
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆40Jan 16, 2026Updated last month
- ☆279Aug 20, 2022Updated 3 years ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆237Nov 24, 2025Updated 3 months ago
- ☆92Feb 16, 2026Updated 3 weeks ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstacles☆90Jan 13, 2026Updated last month
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆82Mar 9, 2023Updated 3 years ago
- ☆20Dec 3, 2025Updated 3 months ago
- A simple, performant and scalable JAX-based world modeling codebase.☆143Jan 15, 2026Updated last month
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆263Oct 31, 2025Updated 4 months ago
- ☆317Jan 23, 2022Updated 4 years ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆233Feb 26, 2026Updated last week
- Source files to replicate experiments in my ICLR 2022 paper.☆71Jul 17, 2025Updated 7 months ago