torch
torchvision
mmengine
swanlab