torch
numpy
torchvision
diffusers
transformers
accelerate
ftfy
modin[all]