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