DHDev0 / Stochastic-muzero
Pytorch Implementation of Stochastic MuZero for gym environment. This algorithm is capable of supporting a wide range of action and observation spaces, including both discrete and continuous variations.
☆50Updated 10 months ago
Related projects: ⓘ
- Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observ…☆25Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.☆46Updated 5 months ago
- Experimentation with Regularized Nash Dynamics on a GPU accelerated game☆37Updated last year
- Implementation of MuZero with PyTorch, based on the pseudocode from DeepMind (https://arxiv.org/src/1911.08265v2/anc/pseudocode.py).☆29Updated 2 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆37Updated 2 years ago
- Clean baseline implementation of PPO using an episodic TransformerXL memory☆137Updated 3 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆102Updated 3 weeks ago
- Adaptable tools to make reinforcement learning and evolutionary computation algorithms.☆53Updated 2 years ago
- Datasets with baselines for offline multi-agent reinforcement learning.☆125Updated this week
- MiniZero: An AlphaZero and MuZero Training Framework☆63Updated last month
- A collection of Deep Reinforcement Learning algorithms implemented with PyTorch to solve Atari games and classic control tasks like CartP…☆99Updated 6 months ago
- Classic MCTS example with mctx☆15Updated last year
- ☆15Updated 3 months ago
- Official Implementation of "Can Learned Optimization Make Reinforcement Learning Less Difficult"☆10Updated 2 months ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆189Updated 2 weeks ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆76Updated 3 months ago
- ☆59Updated last month
- Prioritized Experience Replay implementation with proportional prioritization☆67Updated last year
- 🤖 Elegant implementations of offline safe RL algorithms in PyTorch☆161Updated this week
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆96Updated 2 years ago
- fast + parallel AlphaZero in JAX☆80Updated 5 months ago
- Implementation of Trajectory Transformer with attention caching and batched beam search☆101Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆78Updated last year
- ☆46Updated last year
- Baseline implementation of recurrent PPO using truncated BPTT☆118Updated 4 months ago
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆70Updated last year
- PyTorch Implementation of Implicit Quantile Networks (IQN) for Distributional Reinforcement Learning with additional extensions like PER,…☆77Updated last year
- An API conversion tool for popular external reinforcement learning environments☆131Updated 3 months ago
- Deep Transformer Q-Networks for Partially Observable Reinforcement Learning☆130Updated 2 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆87Updated last month