accelerate
beartype
einops>=0.8.1
torch>=2.4
