toshikwa / rljax
A collection of RL algorithms written in JAX.
☆95Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for rljax
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Benchmarking RL generalization in an interpretable way.☆132Updated 9 months ago
- Revisiting Rainbow☆73Updated 3 years ago
- Datasets for data-driven deep reinforcement learning with PyBullet environments☆143Updated 3 years ago
- Datasets for data-driven deep reinforcement learning with Atari (wrapper for datasets released by Google)☆107Updated 2 months ago
- Baselines for gymnax 🤖☆60Updated last year
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆105Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- On the model-based stochastic value gradient for continuous reinforcement learning☆55Updated last year
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆99Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆133Updated last year
- ☆36Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆105Updated 2 months ago
- PyTorch implementation of Stochastic Latent Actor-Critic(SLAC).☆87Updated 3 months ago
- impact-driven-exploration☆128Updated last year
- Deep Hierarchical Planning from Pixels☆90Updated last year
- PAIRED in PyTorch 🔥☆56Updated last year
- AGAC: Adversarially Guided Actor-Critic☆47Updated 3 years ago
- General Modules for JAX☆58Updated 3 months ago
- Pytorch implementations of RL algorithms, focusing on model-based, lifelong, reset-free, and offline algorithms. Official codebase for Re…☆98Updated 2 years ago
- Deep Reinforcement Learning Framework done with PyTorch☆30Updated last week
- ☆29Updated 3 years ago
- Implicit Normalizing Flows + Reinforcement Learning☆60Updated 5 years ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆101Updated 3 months ago
- MultiTask Environments for Reinforcement Learning.☆74Updated 2 years ago
- Implementation of Truncated Quantile Critics method for continuous reinforcement learning. https://bayesgroup.github.io/tqc/☆90Updated 3 years ago
- Code for "Learning to Reach Goals via Iterated Supervised Learning"☆76Updated 2 years ago