numpy>=2.1.0
flextype>=0.2.0

[jax]
jax>=0.8.1

[torch]
torch>=2.6.0
