flax>=0.12.2
jax>=0.8.2
jaxlib>=0.8.2
lark>=1.3.1
matplotlib>=3.10.8
networkx>=3.6.1
numpy>=2.4.1
onnx>=1.20.1
onnx2pytorch>=0.5.3
onnxruntime>=1.23.2
optax>=0.2.6
orbax>=0.1.9
seaborn>=0.13.2
torch>=2.10.0
xarray>=2025.12.0

[docs]
build>=1.4.0
furo>=2025.12.19
nbsphinx>=0.9.8
pandoc>=2.4
sphinx>=9.0.4
sphinx-autodoc-typehints>=3.6.1
sphinx-copybutton>=0.5.2
sphinx-design>=0.7.0
twine>=6.2.0

[extra]
grain>=0.2.15
jraph>=0.0.6.dev0
numpyro>=0.19.0
onnx2torch>=1.5.15
optuna>=4.7.0
optuna-dashboard>=0.20.0
pandas>=2.3.3
pyarrow>=23.0.0
polars>=1.37.1
scipy>=1.17.0
scikit-learn>=1.8.0
sympy>=1.14.0

[test]
pytest>=9.0.2
