accelerate
adam-atan2-pytorch
beartype
einx>=0.3.0
einops>=0.8.0
torch>=2.2
tqdm
