torch
torchinfo

[dev]
pytest
ruff
build
