pin ml_dytpes version

mlcommons · Sep 21, 2023 · a7b60fa · a7b60fa
1 parent d54b866
commit a7b60fa
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -115,6 +115,7 @@ jax_core_deps =
   # Todo(kasimbeg): verify if this is necessary after we
   # upgrade jax.
   chex==0.1.7
+  ml_dtypes==0.2.0
 
 # JAX CPU
 jax_cpu =