jaxtyping
numpy
torch

[dev]
basedpyright
pytest
ruff
