Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights on all environments.
☆62Aug 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:
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- ☆18Apr 17, 2026Updated 4 months ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆43Jun 14, 2021Updated 5 years ago
- JAX implementations of various deep reinforcement learning algorithms.☆25Feb 2, 2025Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆758Oct 26, 2022Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Various reinforcement learning algorithms written in Jax + Flax☆26Jun 24, 2023Updated 3 years ago
- RL Environments in JAX 🌍☆914Updated this week
- A collection of RL algorithms written in JAX.☆106Jul 5, 2022Updated 4 years ago
- Collection of resources on plasticity loss in deep reinforcement learning☆23Nov 12, 2024Updated last year
- A2C training of Relational Deep Reinforcement Learning Architecture☆13Jun 22, 2022Updated 4 years ago
- 📴 OffCon^3: SOTA PyTorch SAC and TD3 Implementations (arxiv: 2101.11331)☆25Jun 20, 2021Updated 5 years ago
- ☆58Jan 20, 2023Updated 3 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆281Sep 22, 2025Updated 10 months ago
- POPGym Library in JAX☆14Apr 15, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Action Value Gradient Algorithm☆30May 18, 2025Updated last year
- Revisiting Rainbow☆76Jun 9, 2021Updated 5 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆31Nov 22, 2022Updated 3 years ago
- Modular framework for Reinforcement Learning in python☆185Feb 1, 2023Updated 3 years ago
- Standard interface for entity based reinforcement learning environments.☆40Feb 28, 2024Updated 2 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆418Mar 18, 2026Updated 4 months ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 5 years ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆105Jan 23, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Extending JAX with custom C++ and CUDA code☆404Aug 18, 2024Updated 2 years ago
- ☆13Aug 9, 2022Updated 4 years ago
- ☆16Jul 1, 2021Updated 5 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆607Jul 26, 2026Updated 3 weeks ago
- krazy grid world☆26Mar 2, 2020Updated 6 years ago
- Integrates Imbue's Cost Aware pareto-Region Bayesian Search (CARBS) with Weights and Biases (WanDB)☆12Mar 17, 2025Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆67Jul 25, 2026Updated 3 weeks ago
- Rainbow DQN implementation accompanying the paper "Fast and Data-Efficient Training of Rainbow" which reaches 205.7 median HNS after 10M …☆46Dec 11, 2021Updated 4 years ago
- [AAAI'25 Oral] Are Expressive Models Truly Necessary for Offline RL?☆15Dec 10, 2024Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Simple single-file baselines for Q-Learning in pure-GPU setting☆243Nov 24, 2025Updated 8 months ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,496Jul 17, 2026Updated last month
- Code for the paper "Leveraging Procedural Generation to Benchmark Reinforcement Learning"☆182Apr 2, 2023Updated 3 years ago
- ☆13Aug 28, 2025Updated 11 months ago
- ☆1,436Aug 6, 2026Updated last week
- POMDP wrappers for OpenAI Gym☆15Nov 4, 2019Updated 6 years ago
- Generalised UDRL☆37May 12, 2022Updated 4 years ago