No public description

PiperOrigin-RevId: 650786820
2024-07-09 16:05:18 -07:00
parent 28dcfa669b
commit 179d20df4a
17 changed files with 1241 additions and 334 deletions
@@ -15,3 +15,4 @@ dependencies:
    - paxml
    - jax[cuda12]==0.4.26
    - einshape
+    - scikit-learn