Really Fast End-to-End Jax RL Implementations
β1,065Sep 9, 2024Updated last year
Alternatives and similar repositories for purejaxrl
Users that are interested in purejaxrl are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- RL Environments in JAX πβ887Apr 2, 2026Updated last month
- ποΈA research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX β’ End-to-End JAX RLβ406Mar 18, 2026Updated last month
- Multi-Agent Reinforcement Learning with JAXβ798Apr 30, 2026Updated last week
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β265Oct 31, 2025Updated 6 months ago
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ279Sep 22, 2025Updated 7 months ago
- Deploy on Railway without the complexity - Free Credits Offer β’ AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsβ585Feb 25, 2026Updated 2 months ago
- Accelerated minigrid environments with JAXβ168Oct 20, 2025Updated 6 months ago
- πΉοΈ A diverse suite of scalable reinforcement learning environments in JAXβ828Apr 13, 2026Updated 3 weeks ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.β399Feb 10, 2026Updated 2 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ239Nov 24, 2025Updated 5 months ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid ποΈβ336Dec 16, 2025Updated 4 months ago
- Efficient baselines for autocurricula in JAX.β212Aug 24, 2024Updated last year
- βοΈ Vectorized RL game environments in JAXβ603Mar 6, 2025Updated last year
- Baselines for gymnax π€β76Apr 3, 2023Updated 3 years ago
- GPUs on demand by Runpod - Special Offer Available β’ AdRun AI, ML, and HPC workloads on powerful cloud GPUsβwithout limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Clean single-file implementation of offline RL algorithms in JAXβ175Nov 24, 2025Updated 5 months ago
- β1,422Mar 31, 2026Updated last month
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.β756Oct 26, 2022Updated 3 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]β114Dec 5, 2023Updated 2 years ago
- Evolution Strategies in JAX π¦β755Apr 2, 2026Updated last month
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.β249Feb 26, 2026Updated 2 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learningβ78Aug 18, 2024Updated last year
- High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, Tβ¦β9,679Apr 20, 2026Updated 2 weeks ago
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.β264Nov 23, 2025Updated 5 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.
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.β872Aug 12, 2024Updated last year
- β96Jan 21, 2026Updated 3 months ago
- Partially Observable Process Gymβ214Jun 12, 2025Updated 10 months ago
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.β1,336Updated this week
- π¦ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXβ909Apr 21, 2026Updated 2 weeks ago
- Massively parallel rigidbody physics simulation on accelerator hardware.β3,148Updated this week
- β262Mar 11, 2026Updated last month
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ124Aug 22, 2024Updated last year
- Simple JAX Graphics Library.β37Nov 3, 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.
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.β691Aug 22, 2025Updated 8 months ago
- β94Feb 16, 2026Updated 2 months ago
- JAX - A curated list of resources https://github.com/google/jaxβ2,102Jan 20, 2026Updated 3 months ago
- JAX implementation of RL algorithms and vectorized environmentsβ51Dec 26, 2023Updated 2 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/β2,872Updated this week
- Monte Carlo tree search in JAXβ2,619Sep 2, 2025Updated 8 months ago
- β331Dec 19, 2024Updated last year