numpy>=1.24
scipy>=1.10
click>=8.0

[dev]
pytest>=7.0
ruff>=0.1.0
black>=23.0

[jax]
jax>=0.9.2

[jax-cpu]
jax[cpu]>=0.9.2

[jax-cuda12]
jax[cuda12]>=0.9.2

[jax-cuda13]
jax[cuda13]>=0.9.2

[jax-tpu]
jax[tpu]>=0.9.2

[test]
pytest>=7.0
ruff>=0.1.0
black>=23.0
jax>=0.9.2
e3nn-jax>=0.21.0
e3nn>=0.5.0
