Add ml-dtypes as a dev dependency.

Author: Hilly12

pyproject.toml

@@ -10,6 +10,7 @@ dynamic = ["dependencies"]
 [dependency-groups]
 dev = [
     "einops>=0.8.1",
+    "ml-dtypes>=0.5.1",
     "pre-commit>=4.2.0",
     "pylint>=3.3.6",
     "pytest>=8.3.5",