← all repositories

luchris429/purejaxrl

PureJaxRL is a high-performance end-to-end reinforcement learning implementation library built entirely in JAX.

1.1k stars Python ML Frameworks
purejaxrl
Velocity · 7d
+0.9
★ / day
Trend
steady
star history

The library implements the entire RL training pipeline in JAX, including environments, enabling significant speedups through JIT compilation and parallel execution on GPUs. It supports running many agents in parallel for rapid hyperparameter tuning and meta-evolution for discovering new RL algorithms. The implementation includes PPO and other standard RL algorithms optimized for high-throughput training.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.