numpy!=1.19.3,>1.17.2
torchvision>=0.10.0
torch>=1.9.0
torchmetrics>=0.6.0
typing-extensions>=4.1.1

[mushin]
pytorch-lightning>=1.5.0
hydra-zen>=0.6.0
xarray>=0.19.0
matplotlib>=3.3

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