torch
torchvision
colossalai
