ikostrikov / jaxrlLinks
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
☆723Updated 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.☆848Updated last year
- Library for Model Based RL☆1,024Updated last year
- RL Environments in JAX 🌍☆819Updated 4 months ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆480Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆512Updated last week
- A Python interface for reinforcement learning environments☆379Updated 2 years ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆447Updated this week
- Imitation learning algorithms☆553Updated 7 months ago
- Code for the paper "When to Trust Your Model: Model-Based Policy Optimization"☆512Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆557Updated 4 years ago
- Code for conservative Q-learning☆457Updated 3 years ago
- Simple (but often Strong) Baselines for POMDPs in PyTorch, ICML 2022☆337Updated last year
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆313Updated last year
- ☆354Updated 3 years ago
- Multi-objective Gymnasium environments for reinforcement learning☆346Updated 3 months ago
- Really Fast End-to-End Jax RL Implementations☆963Updated last year
- PyTorch implementation of Soft Actor-Critic (SAC)☆561Updated 3 years ago
- Author's PyTorch implementation of TD3+BC, a simple variant of TD3 for offline RL☆374Updated 3 years ago
- ☆313Updated 10 months ago
- Code for the paper "Phasic Policy Gradient"☆266Updated 2 years ago
- ☆240Updated 11 months ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆229Updated 2 years ago
- Offline Reinforcement Learning (aka Batch Reinforcement Learning) on Atari 2600 games☆556Updated 2 years ago
- Tools for accelerating safe exploration research.☆562Updated 2 years ago
- RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code☆701Updated last year
- Real-World RL Benchmark Suite☆356Updated 5 years ago
- High throughput synchronous and asynchronous reinforcement learning☆946Updated 5 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆764Updated last week
- This is the pytorch implementation of Hindsight Experience Replay (HER) - Experiment on all fetch robotic environments.☆436Updated 3 years ago
- Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPO☆205Updated 3 years ago