beartype
einops>=0.8.0
torch>=2.0
