JAX implementations of various deep reinforcement learning algorithms.
☆25Feb 2, 2025Updated last year
Alternatives and similar repositories for JAX-RL
Users that are interested in JAX-RL are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 4 years ago
- The code for experiments conducted to verify the correctness of mirror learning.☆11Jun 3, 2022Updated 4 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)☆12Jun 15, 2023Updated 3 years ago
- JAX implementation of NeuroEvolution of Augmenting Topologies (Neat)☆11Jul 6, 2024Updated 2 years ago
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 years 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.
- Code for "Meta Learning Backpropagation And Improving It" @ NeurIPS 2021 https://arxiv.org/abs/2012.14905☆33Jan 9, 2022Updated 4 years ago
- A collection of RL algorithms written in JAX.☆107Jul 5, 2022Updated 4 years ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆23Dec 22, 2020Updated 5 years ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆63Aug 4, 2022Updated 4 years ago
- Official implementation of "NinA: Normalizing Flows in Action. Training VLA Models with Normalizing Flows"☆19Sep 22, 2025Updated last year
- 🧮 QuatIca: Advanced Quaternion Linear Algebra Library for Python A comprehensive numerical library for quaternion matrix operations, fe…☆14Mar 26, 2026Updated 7 months ago
- ☆12May 21, 2026Updated 5 months ago
- Autoregressive transformer in JAX from scratch☆23Jan 28, 2022Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- A JAX Implementation of the Twin Delayed DDPG Algorithm☆35Mar 12, 2020Updated 6 years ago
- A High-Throughput JAX-native Environment for Abstraction and Reasoning Research☆17Aug 8, 2026Updated 2 months ago
- A high throughput, end-to-end RL library for infinite-horizon tasks.☆24Mar 24, 2026Updated 7 months ago
- RL Environments in JAX 🌍☆937Updated this week
- Mini RL Lab☆16Jun 17, 2024Updated 2 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆757Oct 26, 2022Updated 4 years ago
- ROS base local planner plugin which uses the tangent bug algorithm☆22May 10, 2020Updated 6 years ago
- See and Tell | Assistant to Experience Video Content With Ears☆20Oct 3, 2023Updated 3 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆44Sep 19, 2022Updated 4 years 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.
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 11 months ago
- Baselines for gymnax 🤖☆82Apr 3, 2023Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆283Jun 10, 2026Updated 4 months ago
- Simple CIFAR10 ResNet example with JAX.☆23Jun 1, 2021Updated 5 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆425Mar 18, 2026Updated 7 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆25Feb 15, 2025Updated last year
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆67Oct 23, 2026Updated last week
- A tiny conversational Minionese language model trained from scratch on Apple Silicon☆17Jul 11, 2026Updated 3 months ago
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆38Jan 8, 2026Updated 9 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.
- ☆17Feb 12, 2025Updated last year
- Least Squares Policy Iteration (LSPI) in Python☆11May 25, 2015Updated 11 years ago
- A general framework for cascade correlation architectures in Python with wrappers to keras, tensorflow and sklearn☆13Jan 7, 2025Updated last year
- 🫧 Differentiable microbubble dynamics in JAX.☆17Oct 7, 2026Updated 3 weeks ago
- A fast Python solver for Markov Decision Processes☆11Oct 9, 2026Updated 3 weeks ago
- Accelerated minigrid environments with JAX☆184Sep 10, 2026Updated last month
- flexible meta-learning in jax☆17Oct 19, 2023Updated 3 years ago