orjson
torch
gluonts>=0.10.0
pytorch-lightning
datasets
xformers
git+https://github.com/ml-jku/hopfield-layers
etsformer-pytorch
reformer_pytorch
einops
opt_einsum
pykeops
scipy
apex
git+https://github.com/microsoft/torchscale
