Update dependencies in pyproject.toml

Added jax (<0.7.0) and seaborn (^0.13.2) to dependencies. Updated optax to ^0.2.5 and equinox to ^0.13 for improved compatibility and features.

Author: JR-1991

pyproject.toml

@@ -10,6 +10,7 @@ readme = "README.md"
 python = ">=3.10,<3.14"
 matplotlib = "^3.8.0"
 dotted-dict = "^1.1.3"
+jax = "<0.7.0"
 sympy = "^1.12"
 pandas = "^2.1.1"
 tqdm = "^4.66.1"
@@ -20,13 +21,14 @@ corner = "^2.2.2"
 lmfit = "^1.2.2"
 jinja2 = "^3.1.2"
 ipywidgets = "^8.1.1"
-optax = "^0.2.3"
+optax = "^0.2.5"
 diffrax = "^0.7.0"
 pydantic = "^2.9.2"
-equinox = "^0.12.2"
+equinox = "^0.13"
 rich = "^13.8.1"
 mlcroissant = "^1.0.10"
 pyenzyme = "^2.0.0"
+seaborn = "^0.13.2"
 
 [tool.setuptools.packages.find]
 where = ["catalax"]