Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.
☆63Aug 4, 2022Updated 4 years ago
Alternatives and similar repositories for ppo_jax
Users that are interested in ppo_jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆19Apr 17, 2026Updated 6 months ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Jun 14, 2021Updated 5 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆759Oct 26, 2022Updated 4 years ago
- Revisiting Rainbow☆76Jun 9, 2021Updated 5 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- RL Environments in JAX 🌍☆940Oct 30, 2026Updated last week
- Accelerated minigrid environments with JAX☆184Sep 10, 2026Updated last month
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- A collection of RL algorithms written in JAX.☆107Jul 5, 2022Updated 4 years ago
- Various reinforcement learning algorithms written in Jax + Flax☆26Oct 24, 2026Updated last week
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆282Aug 27, 2026Updated 2 months ago
- Standard interface for entity based reinforcement learning environments.☆41Feb 28, 2024Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 4 years ago
- A2C training of Relational Deep Reinforcement Learning Architecture☆13Jun 22, 2022Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Baselines for gymnax 🤖☆82Apr 3, 2023Updated 3 years ago
- Code for the paper: Sparsely Changing Latent States for Prediction and Planning in Partially Observable Domains☆12Nov 12, 2021Updated 4 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 2021Updated 5 years ago
- A dataloader, but for JAX☆20May 17, 2024Updated 2 years ago
- Collection of resources on plasticity loss in deep reinforcement learning☆23Nov 12, 2024Updated last year
- Modular framework for Reinforcement Learning in python☆186Feb 1, 2023Updated 3 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆24Nov 8, 2024Updated last year
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆425Mar 18, 2026Updated 7 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 5 years ago
- ☆23Aug 19, 2022Updated 4 years ago
- Code for the paper "Leveraging Procedural Generation to Benchmark Reinforcement Learning"☆182Apr 2, 2023Updated 3 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆32Nov 22, 2022Updated 3 years ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆107Jan 23, 2025Updated last year
- Extending JAX with custom C++ and CUDA code☆402Aug 18, 2024Updated 2 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆615Oct 27, 2026Updated last week
- ☆13Aug 9, 2022Updated 4 years ago
- ☆16Jul 1, 2021Updated 5 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆107May 17, 2022Updated 4 years ago
- A simple, performant and scalable JAX-based world modeling codebase.☆165Jan 15, 2026Updated 9 months ago
- [AAAI'25 Oral] Are Expressive Models Truly Necessary for Offline RL?☆14Dec 10, 2024Updated last year
- Really Fast End-to-End Jax RL Implementations☆1,122Sep 9, 2024Updated 2 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- Code and links for over 25,000 trained Atari agents☆100Aug 22, 2024Updated 2 years ago
- Convert DeepMind Control Suite to OpenAI gym environments.☆87Jan 31, 2020Updated 6 years ago