torch
torchvision
einops
einops-exts
resize-right
typing
transformers
