accelerate>=0.25.0
beartype
huggingface_hub>=0.21.4
classifier-free-guidance-pytorch>=0.6.2
einops>=0.7.0
einx[torch]>=0.1.3
ema-pytorch>=0.5.0
environs
gateloop-transformer>=0.2.2
jaxtyping
local-attention>=1.9.0
numpy
pytorch-custom-utils>=0.0.9
taylor-series-linear-attention>=0.1.6
torch>=2.1
torch_geometric
tqdm
vector-quantize-pytorch>=1.14.22
x-transformers>=1.30.6
