datasets==2.4.0
einops==0.4.1
geotorch==0.3.0
h5py==3.7.0
numpy==1.21.6
regex==2022.6.2
six==1.16.0
torch==1.11.0
torchcubicspline==0.0.3
torchsde==0.2.5
torchvision==0.12.0
tqdm==4.64.0
transformers==4.20.1
wandb==0.12.21
