torch
torchinfo
numpy

[dev]
pytest
ruff
build
