sotetsuk / pgxLinks
♟️ Vectorized RL game environments in JAX
☆510Updated 4 months ago
Alternatives and similar repositories for pgx
Users that are interested in pgx are comparing it to the libraries listed below
Sorting:
- RL Environments in JAX 🌍☆790Updated 2 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆745Updated last month
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆345Updated this week
- Really Fast End-to-End Jax RL Implementations☆925Updated 10 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆330Updated 3 weeks ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆242Updated this week
- Multi-Agent Reinforcement Learning with JAX☆609Updated 3 weeks ago
- Accelerated Quality-Diversity☆313Updated last month
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆488Updated this week
- fast + parallel AlphaZero in JAX☆97Updated 7 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆230Updated 2 months ago
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…☆160Updated 4 years ago
- Modular framework for Reinforcement Learning in python☆174Updated 2 years ago
- Evolution Strategies in JAX 🦎☆651Updated last month
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆699Updated 2 years ago
- AlphaZero in JAX☆78Updated last year
- Distributed Reinforcement Learning accelerated by Lightning Fabric☆382Updated last week
- Benchmarking the Spectrum of Agent Capabilities☆461Updated last year
- Accelerated minigrid environments with JAX☆139Updated last week
- ☆305Updated 7 months ago
- A grid-world game engine for game AI research☆246Updated last year
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆904Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.☆60Updated 8 months ago
- Classic MCTS example with mctx☆19Updated 2 years ago
- A simple and highly efficient RTS-game-inspired environment for reinforcement learning (formerly Gym-MicroRTS)☆261Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆174Updated 4 months ago
- ☆235Updated 8 months ago
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research☆499Updated 5 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆836Updated 11 months ago
- ☆81Updated 9 months ago