torch
torchvision
diffusers
accelerate
scipy