torch
torchvision
safetensors
numpy
einops
typing_extensions
