beartype
einops>=0.7.0
optree
torch>=2.0
x-transformers
