Popular GitHub Repositories for Gumbel Muzero

A Collection of Variational Autoencoders (VAE) in PyTorch.

MuZero

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios (awesome MCTS)

[Pytorch] Generative retrieval model using semantic IDs from "Recommender Systems with Generative Retrieval"

Pytorch Implementation of MuZero

xingtian is a componentized library for the development and verification of reinforcement learning algorithms

Pytorch implementation of stochastically quantized variational autoencoder (SQ-VAE)

Source code for the NAACL 2019 paper "SEQ^3: Differentiable Sequence-to-Sequence-to-Sequence Autoencoder for Unsupervised Abstractive Sentence Compression"

MiniZero: An AlphaZero and MuZero Training Framework

TensorFlow GAN implementation using Gumbel Softmax

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.

Implementation of NeurIPS 20 paper: Latent Template Induction with Gumbel-CRFs

An implementation of MuZero in JAX.

TensorFlow-based implementation of "Attend, Infer, Repeat" paper (Eslami et al., 2016, arXiv:1603.08575).

A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environments.

Pytorch Implementation of MuZero Unplugged for gym environment. This algorithm is capable of supporting a wide range of action and observation spaces, including both discrete and continuous variations.

Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023

Code for TACL 2022 paper on Data-to-text Generation with Variational Sequential Planning

Generalized AI to perform a multitude of tasks written in python3

[ICLR 2025 Oral] OptionZero: A method for autonomously discovering and utilizing options in the MuZero algorithm