diff --git a/pyproject.toml b/pyproject.toml
index 9cfe350..fffc0f2 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -42,7 +42,7 @@ jax-gpu = [
 [dependency-groups]
 lint = ["ruff==0.16.2"]
 typecheck = [
-  "mypy==2.3.0",
+  "mypy==2.3.1",
   "types-tensorflow>=2.18,<3",
   "microsoft-python-type-stubs @ git+https://github.com/microsoft/python-type-stubs.git@main",
 ]