absl-py>=2.0.0
jax>=0.4.31
flax>=0.7.5
numpy>2
immutabledict

[:python_version >= "3.11"]
setuptools

[testing]
chex
jaxonnxruntime
onnx
pytest
pytest-xdist
pylint>=2.6.0
pyink
pyyaml
torch>=2.7.1

[tglfnnukaea]
torch>=2.7.1
pyyaml
