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

PiperOrigin-RevId: 688617061

Author: Google-ML-Automation

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',