assoc-scan>=0.0.2
einops>=0.8
torch>=2.2
