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