the_well>=1.0.0
torch>=2.0.0
einops
wandb
tqdm
h5py
fsspec
aiohttp