pytorch-lightning>=2.4.0
hydra-zen>=0.17.0
hydra-core>=1.3.7
torch>=2.4.0
numpy>=2
omegaconf>=2.2
xarray>=2024.6.0
pandas>=2.2.2

[audio]
mushin-py[eval]

[audio:sys_platform != "win32"]
pystoi>=0.3.3

[detection]
mushin-py[eval]
torchvision>=0.19

[detection:sys_platform != "win32"]
pycocotools>=2.0.7

[eval]
torchmetrics>=1.0
scipy>=1.13

[image]
mushin-py[eval]
torchvision>=0.19
lpips>=0.1.4

[mcp]
mcp<3,>=2

[netcdf]
netCDF4>=1.7.0

[viz]
matplotlib>=3.9
