torch
torchvision
datasets
transformers
av
tensorboard
colossalai
