bmazoure / ppo_jax
Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.
☆49Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for ppo_jax
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- ☆36Updated last year
- Vectorization techniques for fast population-based training.☆54Updated 2 years ago
- Baselines for gymnax 🤖☆60Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆79Updated last year
- ExORL: Exploratory Data for Offline Reinforcement Learning☆105Updated 2 years ago
- Implicit Normalizing Flows + Reinforcement Learning☆60Updated 5 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆99Updated 2 years ago
- Deep Reinforcement Learning Framework done with PyTorch☆30Updated last week
- Evaluating long-term memory of reinforcement learning algorithms☆133Updated last year
- Revisiting Rainbow☆73Updated 3 years ago
- ☆41Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆105Updated 2 months ago
- An implementation of MuZero in JAX.☆53Updated 2 years ago
- ☆29Updated 3 years ago
- Simple implementation of V-MPO proposed in https://arxiv.org/abs/1909.12238☆44Updated 4 years ago
- ☆63Updated 3 months ago
- General Modules for JAX☆58Updated 3 months ago
- AGAC: Adversarially Guided Actor-Critic☆47Updated 3 years ago
- Accelerated replay buffers in JAX☆40Updated 2 years ago
- Episodic Control☆19Updated 2 years ago
- My Body Is A Cage☆38Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆28Updated 2 years ago
- ☆37Updated 2 years ago
- ☆41Updated last month
- Proto-RL: Reinforcement Learning with Prototypical Representations☆82Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆78Updated 2 years ago
- Deep Hierarchical Planning from Pixels☆90Updated last year