accelerate>=0.25.0
beartype
classifier-free-guidance-pytorch>=0.5.1
einops>=0.7.0
ema-pytorch
local-attention>=1.9.0
gateloop-transformer>=0.2.2
numpy
pytorch-custom-utils>=0.0.6
torch>=2.1
torch_geometric
torchtyping
tqdm
vector-quantize-pytorch>=1.12.0
x-transformers>=1.26.0
