accelerate
discrete-continuous-embed-readout
einops>=0.8.2
ema-pytorch
Jinja2
torch>=2.5
torch-einops-utils>=0.0.21

[test]
pytest
x-transformers>=2.15.2
