torch
torchvision
colossalai