jaxtyping
numpy
torch

[dev]
basedpyright
pytest
pytest-lazy-fixtures
ruff
