GoJAX The board game Go implemented in JAX for fast game processing and machine learning training. See also MuZeroGoJAX.