datasets>=4.4.1
h5py>=3.14.0
lightning!=2.6.2,!=2.6.3,>=2.6.0
mlflow>=1.27.0
mpi-list>=0.3
psutil>=7.2.1
pyyaml>=6.0.3
s3fs>=2025.10.0
safetensors>=0.7.0
tensordict>=0.10.0
typer>=0.20.0
xarray>=2024.7.0
zarr>=2.18.2

[cpu]
torch>=2.7.0
torchvision>=0.22.0

[cuda]
torch>=2.7.0
torchvision>=0.22.0
