torch>=2.0
numpy

[dev]
pytest

[hf]
transformers
accelerate

[jax]
jax
equinox
