torch>=1.10.0
torchmetrics>=0.6.0
typing-extensions>=4.1.1

[datasets]
torchvision>=0.10.0

[mushin]
pytorch-lightning>=1.5.0
hydra-zen>=0.9.0
xarray>=0.19.0
matplotlib>=3.3
netCDF4>=1.5.8
protobuf<=3.20.1

[tests]
pytest>=3.8
hypothesis<6.49.0,>=6.41.0
mygrad>=2.0.0
omegaconf>=2.1.1
