torch
torchvision
lightning
torchvision
lightning
einops
wandb

[dev]
pytest
ruff
