jax>=0.4.0
jaxlib>=0.4.0
numpy>=1.20.0

[dev]
pytest
black
ruff

[report]
reportlab>=4.0
optax>=0.2.0

[torch]
torch>=2.0.0
jax2torch>=0.0.7

[units]
pint>=0.20
