test: add stubs to dev deps

pyproject.toml

@@ -36,4 +36,9 @@ dev-dependencies = [
     "ipython>=8.23.0",
     "ipykernel>=6.29.4",
     "mypy>=1.10.0",
+    "jax[cpu]>=0.4.26",
+    "torch>=2.3.0",
+    "pandas-stubs>=2.2.1.240316",
+    "types-pytz>=2024.1.0.20240417",
+    "types-openpyxl>=3.1.0.20240428",
 ]