MooZi is a high-performance general-game-playing system that plans with a learned model. The MooZi core learning algorithm is based on MuZero with a few extensions. MooZi uses JAX eco-system and Ray for the best performance.
MooZi current supports all MinAtar games and some two-players games from OpenSpiel.
MooZi deploys a hierarchial control paradigm to parallizes its workers.
MooZi is shipped with tools to help you montior understand the system.
MooZi Tensorboard hows all important metrics regarding the behavior and performance of the system.
Hidden Space Visualizer
MooZi hidden space visualizer collects hidden states during acting and visualize them either in 3D or 2D.