einops>=0.8.0
torch>=2.2
x-transformers
