frozendict>=2.4.7
numpy>=2.2.6
jax>=0.6.2
setuptools>=80.9.0
jaxlib>=0.6.2
torch>=2.12.0

[mpi-local]
pyses[mpi-system]
mpich>=5.0.1

[mpi-system]
mpi4py>=4.1.1
