torchvision>=0.10.0
torch!=1.9.1,>=1.9.0
torchmetrics>=0.6.0
typing-extensions>=4.1.1

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

[tests]
pytest>=3.8
hypothesis>=6.41.0
mygrad>=2.0.0
omegaconf>=2.1.1
netCDF4>=1.5.8
