flax>=0.12.2
lark>=1.3.1
networkx>=3.6.1
numpy>=1.22.0
matplotlib>=3.5.0
optax>=0.2.3
seaborn>=0.13.2

[cpu]
jax>=0.8.2
jaxlib>=0.8.2

[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

[export]
onnx>=1.20.1
onnxruntime>=1.23.2
onnx2pytorch>=0.5.3
jax2onnx
torch>=2.10.0
xarray>=2025.12.0

[extra]
grain>=0.2.15
jraph>=0.0.6.dev0
numpyro>=0.19.0
orbax-checkpoint>=0.1.9
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

[gpu]
jax[cuda12]>=0.8.2

[test]
pytest>=9.0.2

[tpu]
jax[tpu]>=0.8.2
requests
