mttga / purejaxql
Simple single-file baselines for Q-Learning in pure-GPU setting
☆137Updated 2 months ago
Alternatives and similar repositories for purejaxql:
Users that are interested in purejaxql are comparing it to the libraries listed below
- ☆70Updated 5 months ago
- ☆180Updated 2 months ago
- ☆73Updated 3 months ago
- Baselines for gymnax 🤖☆63Updated last year
- Accelerated minigrid environments with JAX☆128Updated 6 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 5 months ago
- Evaluating long-term memory of reinforcement learning algorithms☆140Updated last year
- Goal-Conditioned Reinforcement Learning with JAX☆120Updated this week
- ☆42Updated 7 months ago
- Reinforcement learning on general 2D physics environments in JAX. ICLR 2025 Oral.☆125Updated this week
- Highly scalable 2D JAX physics engine.☆44Updated 2 weeks ago
- Clean single-file implementation of offline RL algorithms in JAX☆131Updated last month
- General Modules for JAX☆63Updated 6 months ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆70Updated 5 months ago
- Efficient baselines for autocurricula in JAX.☆178Updated 5 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆19Updated 2 months ago
- Benchmarking RL generalization in an interpretable way.☆142Updated this week
- Simple JAX Graphics Library.☆29Updated 3 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆225Updated 3 weeks ago
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆55Updated last year
- Learning diverse options through the Laplacian representation.☆23Updated last year
- ☆213Updated 2 months ago
- JAX implementation of RL algorithms and vectorized environments☆39Updated last year
- An Open-Ended Agentic Simulator☆39Updated 6 months ago
- ☆18Updated 2 weeks ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆275Updated this week
- ☆70Updated 4 months ago
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆48Updated 2 years ago
- a simple and scalable agent for training adaptive policies with sequence-based RL☆110Updated last week
- ExORL: Exploratory Data for Offline Reinforcement Learning☆109Updated 3 years ago