ikostrikov / jaxrlLinks
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆709Updated 2 years ago
Alternatives and similar repositories for jaxrl
Users that are interested in jaxrl are comparing it to the libraries listed below
Sorting:
- Library for Model Based RL☆1,016Updated last year
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆845Updated last year
- RL Environments in JAX 🌍☆801Updated 3 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆500Updated 2 weeks ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆508Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAX☆630Updated 2 months ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆475Updated last year
- A Python interface for reinforcement learning environments☆377Updated 2 years ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆431Updated 3 weeks ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆757Updated 2 months ago
- Really Fast End-to-End Jax RL Implementations☆946Updated last year
- Imitation learning algorithms☆550Updated 5 months ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆329Updated last year
- ☆351Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆552Updated 4 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆308Updated last year
- Code for conservative Q-learning☆454Updated 3 years ago
- PyTorch implementation of Soft Actor-Critic (SAC)☆558Updated 3 years ago
- Tools for accelerating safe exploration research.☆552Updated 2 years ago
- ☆311Updated 8 months ago
- Multi-objective Gymnasium environments for reinforcement learning☆341Updated 2 months ago
- RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code☆702Updated last year
- High throughput synchronous and asynchronous reinforcement learning☆936Updated 3 months ago
- Code for the paper "Phasic Policy Gradient"☆262Updated 2 years ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆373Updated 3 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆227Updated 2 years ago
- ☆237Updated 9 months ago
- Tonic RL library☆417Updated last year
- A collection of reference environments for offline reinforcement learning☆1,569Updated 9 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆348Updated this week