pytorch-lightning>=2.4.0
hydra-zen>=0.10.0
hydra-core>=1.2.0
omegaconf>=2.1.1
xarray>=2024.6.0
typing-extensions>=4.1.0
torchmetrics>=1.0
scipy>=1.10
pandas>=2.2.2

[:sys_platform != "darwin" or platform_machine != "x86_64"]
torch>=2.4.0
numpy>=2

[:sys_platform == "darwin" and platform_machine == "x86_64"]
torch<2.3,>=2.2.0
numpy<2,>=1.24

[mcp]

[mcp:python_version >= "3.10"]
mcp>=1.2

[netcdf]
netCDF4>=1.7.0

[viz]
matplotlib>=3.3
