Merge pull request #24463 from jakevdp:ml-dtypes-version

PiperOrigin-RevId: 688617061
jax-ml · Oct 22, 2024 · 92aa9a3 · 92aa9a3
2 parents c60bafc + 74fc736
commit 92aa9a3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -54,7 +54,7 @@ def load_version_module(pkg_path):
     python_requires='>=3.10',
     install_requires=[
         f'jaxlib >={_minimum_jaxlib_version}, <={_jax_version}',
-        'ml_dtypes>=0.2.0',
+        'ml_dtypes>=0.4.0',
         'numpy>=1.24',
         "numpy>=1.26.0; python_version>='3.12'",
         'opt_einsum',