torch
torchvision
torchaudio
transformers
scikit-image
matplotlib
diffusers
einops