Pin mpmath to <=1.3.0

Without this, this can cause breakages with pytorch nightlies, see pytorch/pytorch#120995

setup.py

@@ -37,7 +37,12 @@
     pass
 
 # Other requirements
-install_requires += ["scipy", "jaxtyping>=0.2.9", "typeguard~=2.13.3"]
+install_requires += [
+    "scipy",
+    "jaxtyping>=0.2.9",
+    "typeguard~=2.13.3",
+    "mpmath>=0.19,<=1.3",  # avoid incompatibiltiy with torch+sympy with mpmath 1.4
+]
 
 
 # Get version