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
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

[audio]
mushin-py[eval]

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

[detection]
mushin-py[eval]

[detection:sys_platform != "darwin" or platform_machine != "x86_64"]
torchvision>=0.19

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

[detection:sys_platform == "darwin" and platform_machine == "x86_64"]
torchvision<0.18,>=0.17

[eval]
torchmetrics>=1.0
scipy>=1.13

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

[image:sys_platform != "darwin" or platform_machine != "x86_64"]
torchvision>=0.19

[image:sys_platform == "darwin" and platform_machine == "x86_64"]
torchvision<0.18,>=0.17

[mcp]
mcp>=1.2

[netcdf]
netCDF4>=1.7.0

[viz]
matplotlib>=3.9
