pydantic>=2.0

[jax]
jax>=0.6.2

[numpy]
numpy

[torch]
torch>=1.11.0
