ikostrikov / jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆607Updated last year
Related projects: ⓘ
- RL Environments in JAX 🌍☆607Updated 2 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆744Updated last month
- Library for Model Based RL☆953Updated 2 months ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆475Updated last year
- A collection of reference environments for offline reinforcement learning☆1,290Updated 7 months ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆450Updated 5 months ago
- SBX: Stable Baselines Jax (SB3 + Jax)☆319Updated 3 weeks ago
- Code for conservative Q-learning☆393Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAX☆394Updated this week
- Dream to Control: Learning Behaviors by Latent Imagination☆506Updated 3 years ago
- Really Fast End-to-End Jax RL Implementations☆671Updated last week
- PyTorch implementation of Soft Actor-Critic (SAC)☆492Updated 2 years ago
- Imitation learning algorithms☆437Updated last month
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆260Updated this week
- ☆328Updated last year
- A Python interface for reinforcement learning environments☆344Updated last year
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆293Updated 3 weeks ago
- Code for the paper "Offline Reinforcement Learning as One Big Sequence Modeling Problem"☆452Updated last year
- Author's PyTorch implementation of BCQ for continuous and discrete actions☆588Updated 3 years ago
- Code for the paper "Phasic Policy Gradient"☆244Updated last year
- Multi-objective Gymnasium environments for reinforcement learning☆270Updated last week
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆277Updated 8 months ago
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games☆526Updated last year
- Reinforcement learning library(framework) designed for PyTorch, implements DQN, DDPG, A2C, PPO, SAC, MADDPG, A3C, APEX, IMPALA ...☆397Updated 3 years ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆311Updated 2 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆203Updated last year
- Implementation of Efficient Off-policy Meta-learning via Probabilistic Context Variables (PEARL)☆471Updated last year
- ☆1,040Updated 3 months ago
- ☆191Updated 7 months ago
- RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code☆643Updated 4 months ago