torch>=2.0
beartype
jaxtyping

[dev]
pytest
ruff
