ikostrikov / jaxrlLinks
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆745Updated 3 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.☆865Updated last year
- Library for Model Based RL☆1,049Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆549Updated last week
- RL Environments in JAX 🌍☆852Updated 8 months ago
- A Python interface for reinforcement learning environments☆390Updated 3 years ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆484Updated 3 weeks ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆492Updated last year
- Really Fast End-to-End Jax RL Implementations☆1,013Updated last year
- Imitation learning algorithms☆559Updated 10 months ago
- Multi-Agent Reinforcement Learning with JAX☆728Updated 2 weeks ago
- Dream to Control: Learning Behaviors by Latent Imagination☆579Updated 4 years ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆529Updated 3 years ago
- Code for conservative Q-learning☆470Updated 4 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆321Updated 2 years ago
- ☆360Updated 3 years ago
- ☆250Updated last year
- A collection of reference environments for offline reinforcement learning☆1,641Updated last year
- PyTorch implementation of Soft Actor-Critic (SAC)☆585Updated 4 years ago
- Multi-objective Gymnasium environments for reinforcement learning☆365Updated last month
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆341Updated last year
- ☆325Updated last year
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games☆559Updated 2 years ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆800Updated 2 months ago
- RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code☆712Updated last year
- Real-World RL Benchmark Suite☆364Updated 5 years ago
- Tools for accelerating safe exploration research.☆580Updated 2 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆236Updated 2 years ago
- Tonic RL library☆419Updated last year
- Implementation of Efficient Off-policy Meta-learning via Probabilistic Context Variables (PEARL)☆506Updated 3 years ago
- Code for the paper "Phasic Policy Gradient"☆267Updated 2 years ago