torch>=2.1.0
einops>=0.7.0

[fla]
fla
