ikostrikov / jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆631Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for jaxrl
- RL Environments in JAX 🌍☆652Updated 4 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆772Updated 3 months ago
- Library for Model Based RL☆963Updated 4 months ago
- A Python interface for reinforcement learning environments☆346Updated last year
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆479Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax)☆343Updated this week
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆457Updated 7 months ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆306Updated 2 months ago
- Multi-Agent Reinforcement Learning with JAX☆441Updated 2 weeks ago
- Really Fast End-to-End Jax RL Implementations☆724Updated 2 months ago
- PyTorch implementation of Soft Actor-Critic (SAC)☆515Updated 2 years ago
- ☆332Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆286Updated 10 months ago
- Imitation learning algorithms☆466Updated 3 months ago
- Dream to Control: Learning Behaviors by Latent Imagination☆513Updated 3 years ago
- Code for the paper "Phasic Policy Gradient"☆252Updated last year
- Tonic RL library☆405Updated 3 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆295Updated 2 weeks ago
- Code for conservative Q-learning☆410Updated 2 years ago
- A collection of reference environments for offline reinforcement learning☆1,348Updated this week
- Multi-objective Gymnasium environments for reinforcement learning☆289Updated 3 weeks ago
- Author's PyTorch implementation of BCQ for continuous and discrete actions☆598Updated 3 years ago
- A collection of MARL benchmarks based on TorchRL☆282Updated last week
- Real-World RL Benchmark Suite☆347Updated 4 years ago
- ☆201Updated 9 months ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆327Updated 2 years ago
- Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPO☆191Updated 2 years ago
- PyTorch implementation of Hierarchical Actor Critic (HAC) for OpenAI gym environments☆295Updated 2 years ago
- Benchmark for Continuous Multi-Agent Robotic Control, based on OpenAI's Mujoco Gym environments.☆331Updated last year
- Implementation of Efficient Off-policy Meta-learning via Probabilistic Context Variables (PEARL)☆472Updated last year