torch<3.0.0,>=2.13.0
torchvision<1.0.0,>=0.25.0
xarray<2026.0.0,>=2025.1.1
submitit<2.0.0,>=1.5.2
diffusers<1.0.0,>=0.38.0
netcdf4<2.0.0,>=1.7.2
wandb<0.29.0,>=0.28.0
tqdm<5.0.0,>=4.67.1
matplotlib<4.0.0,>=3.10.0
hydra-core==1.3.2
axial-attention<0.7.0,>=0.6.1
lightning<3.0.0,>=2.5.0.post0
zarr<3.0.0,>=2.0.0
einops<0.9.0,>=0.8.0
tensordict<0.14.0,>=0.13.0
scipy<2.0.0,>=1.15.1
pyshtools<5.0.0,>=4.13.1
fasteners<0.20,>=0.19
timm<2.0.0,>=1.0.13
dask<2025.0.0,>=2024.12.1
seaborn<0.14.0,>=0.13.2
gcsfs<2026.0.0,>=2025.5.1
huggingface-hub<2.0.0,>=0.34.0
