ase>=3.24.0
clu>=0.0.12
e3x>=1.0.2
einops>=0.8.0
flax>=0.10.2
jax>=0.4.38
lazy-loader>=0.4
numpy>=1.23
optax>=0.2.4
orbax-checkpoint>=0.11.0
pydantic>=2.10.5
tensorflow>=2.12.0
typer>=0.13.0
uncertainty-toolbox>=0.1.1
vesin>=0.3.0
znh5md>=0.4.2

[cuda]
jax[cuda]

[docs]
sphinx>=7.2.6
sphinx-rtd-theme>=1.3.0
myst-parser>=2.0.0
sphinx-copybutton>=0.5.2
furo>=2023.9.10
nbsphinx>=0.9.3
nbsphinx-link>=1.3.0
ipykernel>=6.29.3
pandoc>=2.3

[ipsuite]
ipsuite>=0.2.3

[mlflow]
mlflow>=2.19.0
