torch
torchdiffeq
torchsde
torchvision
omegaconf
einops
open-clip-torch
transformers
safetensors
pytorch_lightning
aiohttp
accelerate

