colossalai
timm
torch
torchvision
tqdm