jax>=0.4.0
jaxlib>=0.4.0
equinox>=0.11.0
e3nn-jax>=0.20.0
numpy>=1.21.0
skala>=1.0.0

[dev]
pytest
pytest-cov
ruff
torch>=2.0.0
e3nn>=0.5.0
opt_einsum_fx>=0.1.4
huggingface_hub>=0.20.0
