mu_zero/trainer2 library