numpy~=1.19.5
torch~=1.11.0
scipy~=1.6.3
scikit-learn~=0.24.2
matplotlib~=3.4.2
wandb~=0.12.17
h5py~=3.1.0
torchmetrics~=0.7.2
pytest~=6.2.4
setuptools~=59.5.0
pandas~=1.3.1
pip~=21.1.2
pytorch-lightning==1.5.10
jsonargparse[signatures]~=4.3.0
hydra-core~=1.1.1
pre-commit
omegaconf~=2.1.1
pynvml~=11.4.1
functorch~=0.1.0
tueplots