assoc-scan>=0.0.2
einx>=0.3.0
einops>=0.8.0
vector-quantize-pytorch>=1.29.0
torch>=2.5
torch-einops-utils>=0.0.32
x-transformers

[examples]
tqdm

[test]
pytest
