torch>=2.4

[jax]
jax>=0.4.20
numpy>=1.23

[test]
pytest>=7
numpy>=1.23
