/MuZeroGoJax

Mu Zero Go implemented with JAX and GoJAX

Primary LanguagePythonMIT LicenseMIT

MuZeroGoJax

MuZero Go implemented with GoJAX.

Tech Stack

Tech stack
Manager
Metrics
Train
Losses
Data
Game
Models
GoJax
Drive

Update step

update step diagram