diff --git a/pyproject.toml b/pyproject.toml
index 5ef929873..1dd37d842 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -144,6 +144,8 @@ filterwarnings = [
     "ignore:.*pkg_resources is deprecated as an API.*:DeprecationWarning",
     # DeprecationWarning: Deprecated call to `pkg_resources.declare_namespace('google')`.
     "ignore:.*Deprecated call to.*pkg_resources.declare_namespace.*:DeprecationWarning",
+    # DeprecationWarning: jax.tree_map is deprecated: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).
+    "ignore:.*jax.tree_map is deprecated.*:DeprecationWarning",
 ]
 
 [tool.coverage.report]