ikostrikov / jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆670Updated 2 years ago
Alternatives and similar repositories for jaxrl:
Users that are interested in jaxrl are comparing it to the libraries listed below
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆425Updated last week
- RL Environments in JAX 🌍☆730Updated 8 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆817Updated 7 months ago
- Library for Model Based RL☆989Updated 8 months ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆487Updated 2 years ago
- Really Fast End-to-End Jax RL Implementations☆845Updated 6 months ago
- A Python interface for reinforcement learning environments☆362Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAX☆549Updated 3 weeks ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆371Updated last week
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆470Updated 11 months ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆299Updated last year
- PyTorch implementation of Soft Actor-Critic (SAC)☆535Updated 3 years ago
- Code for conservative Q-learning☆428Updated 3 years ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆314Updated 7 months ago
- Multi-objective Gymnasium environments for reinforcement learning☆312Updated last month
- ☆339Updated 2 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆216Updated last year
- This is the pytorch implementation of Hindsight Experience Replay (HER) - Experiment on all fetch robotic environments.☆425Updated 3 years ago
- A collection of reference environments for offline reinforcement learning☆1,453Updated 4 months ago
- Dream to Control: Learning Behaviors by Latent Imagination☆527Updated 3 years ago
- Code for the paper "Phasic Policy Gradient"☆259Updated 2 years ago
- Imitation learning algorithms☆507Updated last week
- Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...☆407Updated 3 years ago
- ☆218Updated 4 months ago
- Benchmark for Continuous Multi-Agent Robotic Control, based on OpenAI's Mujoco Gym environments.☆347Updated 2 years ago
- Author's PyTorch implementation of BCQ for continuous and discrete actions☆622Updated 3 years ago
- A collection of MARL benchmarks based on TorchRL☆372Updated this week
- Implementation of Efficient Off-policy Meta-learning via Probabilistic Context Variables (PEARL)☆486Updated 2 years ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆351Updated 3 years ago
- An extension of the PyMARL codebase that includes additional algorithms and environment support☆584Updated 6 months ago