Home Projects purejaxrl
purejaxrl
Python

purejaxrl

Really Fast End-to-End Jax RL Implementations

by luchris429 · GitHub
Stars
Forks
License
Created
Last commit
Language
deep-reinforcement-learningjaxppoApache-2.0Python
View on GitHub
In plain words

Run reinforcement learning experiments much faster by using JAX for the entire training process.

You maintain this project?

Claim its page: indexed whatever its rank, translated into six languages, and enriched with what you write yourself.

Claim this page →
purejaxrl — GitHub preview card
📈 Star history
1 0921 091
2026-07-202026-08-31
📈 Track purejaxrl

Get an email alert on its next release or when it starts trending — never miss the moment.

Free · no card · unsubscribe anytime
Get email alerts →
📄 About

Really Fast End-to-End Jax RL Implementations

purejaxrl has 1.1k stars on GitHub. It has been forked 88 times. purejaxrl is written mainly in Python. It has been in active development since 2023. purejaxrl is available under the Apache-2.0 license. Its main topics are deep-reinforcement-learning, jax, ppo, reinforcement-learning.

Frequently asked questions

What is purejaxrl?

Really Fast End-to-End Jax RL Implementations

Is purejaxrl open source?

purejaxrl is an open-source project. It is released under the Apache-2.0 license.

Is purejaxrl free?

Yes. purejaxrl is free and open source — you can use, modify and self-host it.

What license does purejaxrl use?

purejaxrl is available under the Apache-2.0 license.

What language is purejaxrl written in?

purejaxrl is written mainly in Python.

🏅 Maintainer of this project?
olud.ai badge — purejaxrl

Add this live badge to your README — your GitHub stars and directory rank, refreshed daily.

[![olud.ai](https://olud.ai/badge.php?tool=luchris429-purejaxrl)](https://olud.ai/project/luchris429-purejaxrl.html)
More badge options →
🧬 Shares DNA with🧬 View the DNA map →
PPO-for-Beginners
A simple and well styled PPO implementation. Based on my Medium series: https://medium.com/@eyy…
1.3k · machine-learning
sharespporeinforcement-learning
hands-on-rl
Free course that takes you from zero to Reinforcement Learning PRO 🦸🏻‍🦸🏽
1.6k · deep-reinforcement-learning
sharesdeep-reinforcement-learningreinforcement-learning
PPO-PyTorch
Minimal implementation of clipped objective Proximal Policy Optimization (PPO) in PyTorch
2.4k · deep-learning
sharesppodeep-reinforcement-learning
drl-zh
Deep Reinforcement Learning: Zero to Hero!
2.3k · deep-learning
sharesdeep-reinforcement-learningreinforcement-learning
reward-learning-rl
[RSS 2019] End-to-End Robotic Reinforcement Learning without Reward Engineering
374 · deep-learning
sharesdeep-reinforcement-learningreinforcement-learning
Awesome-RL-for-Video-Generation
A curated list of papers on reinforcement learning for video generation
565 · dpo
sharespporeinforcement-learning
mctx
Monte Carlo tree search in JAX
2.6k · jax
sharesjaxreinforcement-learning
jumanji
🕹️ A diverse suite of scalable reinforcement learning environments in JAX
851 · jax
sharesjaxreinforcement-learning
rl-collision-avoidance
Implementation of the paper "Towards Optimally Decentralized Multi-Robot Collision Avoidance v…
458 · collision-avoidance
sharespporeinforcement-learning
MARL-Algorithms
Implementations of IQL, QMIX, VDN, COMA, QTRAN, MAVEN, CommNet, DyMA-CL, and G2ANet on SMAC, t…
1.8k · deep-reinforcement-learning
sharesdeep-reinforcement-learningreinforcement-learning

Measured from GitHub topics shared by both projects, weighted by how rare each topic is.