Update README.md

zpyoung · Jul 29, 2022 · eccb96a · eccb96a
1 parent 02db55a
commit eccb96a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,5 @@
 # a0-jax
-AlphaZero in JAX using PAX library.
+AlphaZero in JAX using deepmind [mctx](https://github.com/deepmind/mctx) library.
 
 ```sh
 pip install -r requirements.txt
@@ -94,4 +94,4 @@ python play.py
 
 ## TPU sponsor
 
-Agents in the above demos are trained on Google TPUs sponsored by Google under the [TPU Research Cloud program](https://sites.research.google/trc/about/). 
+Agents in the above demos are trained on Google TPUs sponsored by Google under the [TPU Research Cloud program](https://sites.research.google/trc/about/).