torch>=1.9.0
torchvision
einops
numpy
scipy

[flash]
flash-attn>=2.0.0
xformers
