beartype
einops>=0.8.0
packaging>=21.0
torch>=2.3
tqdm
