
[config]
pyyaml>=5.4

[dev]
numpy>=1.21
pyyaml>=5.4
pytest>=7.0

[jax]
jax>=0.4
optax>=0.1

[numpy]
numpy>=1.21

[tensorflow]
tensorflow>=2.12

[torch]
torch>=2.0
