hr0nix / dejaxView external linksLinks
Accelerated replay buffers in JAX
β46Sep 17, 2022Updated 3 years ago
Alternatives and similar repositories for dejax
Users that are interested in dejax are comparing it to the libraries listed below
Sorting:
- πͺ The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAXβ61Oct 23, 2023Updated 2 years ago
- Modular Single-file Reinfocement Learning Algorithms Libraryβ38May 16, 2023Updated 2 years ago
- Baselines for gymnax π€β74Apr 3, 2023Updated 2 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ56May 21, 2023Updated 2 years ago
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.β19Aug 4, 2023Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gatingβ20Oct 19, 2022Updated 3 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.β15Jan 3, 2023Updated 3 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learningβ22Sep 4, 2023Updated 2 years ago
- General Modules for JAXβ72Sep 12, 2025Updated 5 months ago
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ270Sep 22, 2025Updated 4 months ago
- Docker containers of baseline agents for the Crafter environmentβ30Dec 14, 2021Updated 4 years ago
- 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
- Reinforcement learning training framework for entity-gym environments.β17Mar 18, 2024Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"β104May 17, 2022Updated 3 years ago
- Simple JAX Graphics Library.β36Nov 3, 2024Updated last year
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ122Aug 22, 2024Updated last year
- Vectorization techniques for fast population-based training.β57Aug 12, 2022Updated 3 years ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.β371Jul 7, 2025Updated 7 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Juliaβ23Feb 15, 2025Updated last year
- An Open-Ended Agentic Simulatorβ58Aug 11, 2024Updated last year
- RL Environments in JAX πβ857May 30, 2025Updated 8 months ago
- A tool for aggregating and plotting MARL experiment data.β82Jan 26, 2026Updated 2 weeks ago
- β13Aug 9, 2022Updated 3 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β260Oct 31, 2025Updated 3 months ago
- ποΈA research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX β’ End-to-End JAX RLβ390Oct 29, 2025Updated 3 months ago
- JMP is a Mixed Precision library for JAX.β211Jan 30, 2025Updated last year
- OpenAi's gym environment wrapper to vectorize them with Rayβ23May 25, 2023Updated 2 years ago
- PushWorld: A benchmark for manipulation planning with tools and movable obstaclesβ90Jan 13, 2026Updated last month
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributionsβ73Aug 31, 2024Updated last year
- A collection of matrix games in JAXβ13Nov 28, 2024Updated last year
- Fast reinforcement learning researchβ61Dec 7, 2024Updated last year
- β48Nov 20, 2025Updated 2 months ago
- Accelerated minigrid environments with JAXβ156Oct 20, 2025Updated 3 months ago
- β91Jan 21, 2026Updated 3 weeks ago
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.β100Jul 5, 2023Updated 2 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024β28Aug 9, 2024Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).β17Aug 8, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafterβ18Nov 29, 2025Updated 2 months ago
- Actor-Sharer-Learner training framework for off-policy DRL algorithmsβ22Dec 29, 2024Updated last year