beartype
einops>=0.8.0
torch>=2.4
