torch
transformers
accelerate
pillow
einops
timm
opencv-python