torch
torchvision
lightning
wandb

[dev]
pytest
ruff
torchinfo
