ikostrikov / jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆659Updated 2 years ago
Alternatives and similar repositories for jaxrl:
Users that are interested in jaxrl are comparing it to the libraries listed below
- RL Environments in JAX 🌍☆709Updated 7 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆807Updated 6 months ago
- Really Fast End-to-End Jax RL Implementations☆823Updated 5 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆401Updated this week
- Multi-Agent Reinforcement Learning with JAX☆520Updated this week
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆346Updated last week
- Library for Model Based RL☆986Updated 7 months ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆483Updated 2 years ago
- Multi-objective Gymnasium environments for reinforcement learning☆306Updated last month
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆465Updated 10 months ago
- Imitation learning algorithms☆497Updated 7 months ago
- ☆215Updated 3 months ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆295Updated last year
- A collection of reference environments for offline reinforcement learning☆1,420Updated 3 months ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆311Updated 6 months ago
- Dream to Control: Learning Behaviors by Latent Imagination☆525Updated 3 years ago
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games☆544Updated last year
- Code for conservative Q-learning☆424Updated 3 years ago
- ☆339Updated 2 years ago
- A collection of MARL benchmarks based on TorchRL☆350Updated this week
- Code for the paper "Phasic Policy Gradient"☆259Updated last year
- ☆1,126Updated 9 months ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆342Updated 3 years ago
- A Python interface for reinforcement learning environments☆356Updated 2 years ago
- PyTorch implementation of Soft Actor-Critic (SAC)☆527Updated 3 years ago
- An extension of the PyMARL codebase that includes additional algorithms and environment support☆566Updated 5 months ago
- Benchmark for Continuous Multi-Agent Robotic Control, based on OpenAI's Mujoco Gym environments.☆344Updated last year
- Basic constrained RL agents used in experiments for the "Benchmarking Safe Exploration in Deep Reinforcement Learning" paper.☆405Updated last year
- Author's PyTorch implementation of BCQ for continuous and discrete actions☆610Updated 3 years ago
- Softlearning is a reinforcement learning framework for training maximum entropy policies in continuous domains. Includes the official imp…☆1,270Updated last year