accelerate>=0.25.0
beartype
huggingface_hub>=0.21.4
classifier-free-guidance-pytorch>=0.6.10
einops>=0.8.0
einx[torch]>=0.3.0
ema-pytorch>=0.5.1
environs
gateloop-transformer>=0.2.2
jaxtyping
local-attention>=1.9.9
numpy
pytorch-custom-utils>=0.0.9
rotary-embedding-torch>=0.6.4
taylor-series-linear-attention>=0.1.6
torch>=2.1
torch_geometric
tqdm
vector-quantize-pytorch>=1.14.22
x-transformers<1.31,>=1.30.19
