torch
einops
diffusers
