ikostrikov / jaxrl
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆678Updated 2 years ago
Alternatives and similar repositories for jaxrl
Users that are interested in jaxrl are comparing it to the libraries listed below
Sorting:
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆829Updated 9 months ago
- Library for Model Based RL☆997Updated 10 months ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆489Updated 2 years ago
- RL Environments in JAX 🌍☆744Updated 10 months ago
- A Python interface for reinforcement learning environments☆364Updated 2 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆440Updated 2 weeks ago
- Really Fast End-to-End Jax RL Implementations☆870Updated 8 months ago
- Code for conservative Q-learning☆438Updated 3 years ago
- ☆343Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAX☆575Updated this week
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆359Updated 3 years ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆382Updated last week
- Dream to Control: Learning Behaviors by Latent Imagination☆536Updated 3 years ago
- Imitation learning algorithms☆525Updated last month
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆318Updated 8 months ago
- PyTorch implementation of Soft Actor-Critic (SAC)☆542Updated 3 years ago
- A collection of reference environments for offline reinforcement learning☆1,492Updated 5 months ago
- Softlearning is a reinforcement learning framework for training maximum entropy policies in continuous domains. Includes the official imp…☆1,295Updated last year
- Multi-objective Gymnasium environments for reinforcement learning☆318Updated 2 months ago
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games☆546Updated last year
- Code for the paper "Phasic Policy Gradient"☆261Updated 2 years ago
- Author's PyTorch implementation of BCQ for continuous and discrete actions☆628Updated 4 years ago
- This is the pytorch implementation of Hindsight Experience Replay (HER) - Experiment on all fetch robotic environments.☆431Updated 3 years ago
- ☆1,174Updated 11 months ago
- RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code☆685Updated last year
- ☆228Updated 5 months ago
- Implementation of Efficient Off-policy Meta-learning via Probabilistic Context Variables (PEARL)☆488Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆302Updated last year
- An elegant PyTorch offline reinforcement learning library for researchers.☆329Updated last year
- PyTorch implementation of Hierarchical Actor Critic (HAC) for OpenAI gym environments☆315Updated 3 years ago