tuero / muzero-cpp
A C++ pytorch implementation of MuZero
☆37Updated 11 months ago
Alternatives and similar repositories for muzero-cpp:
Users that are interested in muzero-cpp are comparing it to the libraries listed below
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆40Updated 2 years ago
- An implementation of MuZero in JAX.☆56Updated 2 years ago
- MiniZero: An AlphaZero and MuZero Training Framework☆85Updated last month
- Accelerated replay buffers in JAX☆41Updated 2 years ago
- ☆74Updated 4 months ago
- Deep Reinforcement Learning Framework done with PyTorch☆34Updated 3 weeks 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…☆21Updated 4 years ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆48Updated last week
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and obser…☆64Updated last year
- cfrx is a collection of algorithms and tools for hardware-accelerated Counterfactual Regret Minimization (CFR) algorithms in Jax.☆30Updated 7 months ago
- ☆50Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.☆58Updated 4 months ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆12Updated last year
- Implementation of Diversity Is All You Need (DIAYN) on top of Stable Baselines 3.☆12Updated 2 years ago
- General Modules for JAX☆64Updated last month
- Vectorization techniques for fast population-based training.☆55Updated 2 years ago
- A2C is a special case of PPO!☆19Updated 2 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆53Updated 2 years ago
- AlphaZero in JAX☆77Updated 11 months ago
- Classic MCTS example with mctx☆18Updated last year
- ☆20Updated 9 months ago
- Rainbow DQN implementation accompanying the paper "Fast and Data-Efficient Training of Rainbow" which reaches 205.7 median HNS after 10M …☆44Updated 3 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆50Updated last year
- Reinforcement Learning Assembly☆92Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 7 months ago
- Code of the paper: Debiasing Meta-Gradient Reinforcement Learning by Learning the Outer Value Function☆13Updated 2 years ago
- fast + parallel AlphaZero in JAX☆94Updated 3 months ago
- PyTorch implementation of the Munchausen Reinforcement Learning Algorithms M-DQN and M-IQN☆45Updated 4 years ago
- A PyTorch implementation of SEED, originally created by Google Research for TensorFlow 2.☆13Updated 4 years ago