Simple JAX Graphics Library.
☆36Nov 3, 2024Updated last year
Alternatives and similar repositories for JaxGL
Users that are interested in JaxGL are comparing it to the libraries listed below
Sorting:
- Highly scalable 2D JAX physics engine.☆63Feb 20, 2026Updated last week
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆230Jan 24, 2026Updated last month
- ☆93Jan 21, 2026Updated last month
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆20Oct 21, 2025Updated 4 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆237Nov 24, 2025Updated 3 months ago
- Efficient baselines for autocurricula in JAX.☆208Aug 24, 2024Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆247Nov 23, 2025Updated 3 months ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- ☆19Mar 1, 2023Updated 3 years ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆101Jan 23, 2025Updated last year
- ☆13Apr 25, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 3 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆374Feb 10, 2026Updated 3 weeks ago
- Platform to run interactive Reinforcement Learning agents in a Minecraft Server☆56Feb 20, 2026Updated last week
- XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025☆81Feb 13, 2025Updated last year
- JAX-accelerated Meta-Reinforcement Learning Environments Inspired by XLand and MiniGrid 🏎️☆325Dec 16, 2025Updated 2 months ago
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 2 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 2024Updated last year
- ☆16Jul 16, 2024Updated last year
- ☆23Aug 19, 2022Updated 3 years ago
- An Open-Ended Agentic Simulator☆60Aug 11, 2024Updated last year
- A System for Morphology-Task Generalization via Unified Representation and Behavior Distillation (ICLR2023)☆14Feb 3, 2023Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆37May 19, 2023Updated 2 years ago
- A framework for creating rich, 3D, Minecraft-like single and multi-agent environments for AI research. (Accepted at ICML 2025).☆159Feb 17, 2026Updated 2 weeks ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆395Feb 19, 2026Updated 2 weeks ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Feb 20, 2026Updated last week
- Jax/Flax Implementation of TD-MPC2☆72Jan 12, 2026Updated last month
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 2 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆273Sep 22, 2025Updated 5 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆79Nov 14, 2023Updated 2 years ago
- Bipedal Skills Benchmark for Reinforcement Learning☆25Oct 27, 2022Updated 3 years ago