đī¸A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX âĸ End-to-End JAX RL
â405Mar 18, 2026Updated last month
Alternatives and similar repositories for Stoix
Users that are interested in Stoix are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ⥠Flashbax: Accelerated Replay Buffers in JAXâ278Sep 22, 2025Updated 7 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!â265Oct 31, 2025Updated 6 months ago
- Really Fast End-to-End Jax RL Implementationsâ1,065Sep 9, 2024Updated last year
- RL Environments in JAX đâ886Apr 2, 2026Updated last month
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.â399Feb 10, 2026Updated 2 months 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.
- Accelerated minigrid environments with JAXâ168Oct 20, 2025Updated 6 months ago
- Simple single-file baselines for Q-Learning in pure-GPU settingâ239Nov 24, 2025Updated 5 months ago
- Clean single-file implementation of offline RL algorithms in JAXâ175Nov 24, 2025Updated 5 months ago
- đšī¸ A diverse suite of scalable reinforcement learning environments in JAXâ828Apr 13, 2026Updated 3 weeks ago
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid đī¸â334Dec 16, 2025Updated 4 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.â245Feb 26, 2026Updated 2 months ago
- Multi-Agent Reinforcement Learning with JAXâ798Updated this week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsâ583Feb 25, 2026Updated 2 months ago
- â96Jan 21, 2026Updated 3 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer âĸ AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Efficient baselines for autocurricula in JAX.â212Aug 24, 2024Updated last year
- âī¸ Vectorized RL game environments in JAXâ602Mar 6, 2025Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.â264Nov 23, 2025Updated 5 months ago
- đĻ A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAXâ906Apr 21, 2026Updated last week
- JAX implementation of RL algorithms and vectorized environmentsâ51Dec 26, 2023Updated 2 years ago
- đĒ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXâ62Oct 23, 2023Updated 2 years ago
- A tool for aggregating and plotting MARL experiment data.â84Apr 13, 2026Updated 3 weeks ago
- Simple JAX Graphics Library.â37Nov 3, 2024Updated last year
- A collection of matrix games in JAXâ13Apr 13, 2026Updated 3 weeks 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.
- Simple single file implementations of Reinforcement Learning algorithms in Juliaâ23Feb 15, 2025Updated last year
- Highly scalable 2D JAX physics engine.â65Apr 20, 2026Updated 2 weeks ago
- A framework for Reinforcement Learning research.â255Apr 27, 2026Updated last week
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLâ124Aug 22, 2024Updated last year
- â94Feb 16, 2026Updated 2 months ago
- Baselines for gymnax đ¤â76Apr 3, 2023Updated 3 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)â12Jun 15, 2023Updated 2 years ago
- Evolution Strategies in JAX đĻâ751Apr 2, 2026Updated last month
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.â756Oct 26, 2022Updated 3 years ago
- 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.
- Parallel hyperparameter tuning with JAXâ39Apr 16, 2026Updated 2 weeks ago
- Accelerated replay buffers in JAXâ46Sep 17, 2022Updated 3 years ago
- Datasets with baselines for Offline MARL.â214Nov 2, 2025Updated 6 months ago
- A neural network library written in jaxâ13Feb 3, 2025Updated last year
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"â105Jan 23, 2025Updated last year
- â1,420Mar 31, 2026Updated last month
- Contains JAX implementation of algorithms for inverse reinforcement learningâ78Aug 18, 2024Updated last year