torch>=2.3.0
numpy>=1.26.0
pydantic>=2.7.0
pyyaml>=6.0.1

[bench]
pytest-benchmark>=4.0.0

[dev]
pytest>=8.2.0
pytest-cov>=5.0.0
ruff>=0.4.0
mypy>=1.10.0

[jax]
jax>=0.4.25
jaxlib>=0.4.25
