sotetsuk / pgx
βοΈ Vectorized RL game environments in JAX
β419Updated last week
Alternatives and similar repositories for pgx:
Users that are interested in pgx are comparing it to the libraries listed below
- RL Environments in JAX πβ669Updated 5 months ago
- Really Fast End-to-End Jax RL Implementationsβ754Updated 3 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.β216Updated 2 months ago
- Multi-Agent Reinforcement Learning with JAXβ458Updated this week
- ποΈA research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX β’ End-to-End JAX RLβ251Updated last week
- β‘ Flashbax: Accelerated Replay Buffers in JAXβ216Updated last week
- πΉοΈ A diverse suite of scalable reinforcement learning environments in JAXβ659Updated last week
- Modular framework for Reinforcement Learning in pythonβ168Updated last year
- β158Updated this week
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each otheβ¦β156Updated 3 years ago
- SBX: Stable Baselines Jax (SB3 + Jax)β357Updated last week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.β640Updated 2 years ago
- Accelerated Quality-Diversityβ270Updated 2 months ago
- β206Updated 3 weeks ago
- fast + parallel AlphaZero in JAXβ87Updated 3 weeks ago
- Evolution Strategies in JAX π¦β513Updated last month
- Pytorch Implementation of MuZeroβ345Updated last year
- AlphaZero in JAXβ72Updated 8 months ago
- β288Updated last year
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Researchβ486Updated 3 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.β874Updated 11 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilitiesβ310Updated last week
- A suite of test scenarios for multi-agent reinforcement learning.β636Updated this week
- Distributed Reinforcement Learning accelerated by Lightning Fabricβ326Updated last week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.β785Updated 4 months ago
- Efficient baselines for autocurricula in JAX.β174Updated 3 months ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.β126Updated 3 months ago
- Benchmarking the Spectrum of Agent Capabilitiesβ396Updated 10 months ago
- A Python interface for reinforcement learning environmentsβ348Updated last year
- Partially Observable Process Gymβ168Updated 5 months ago