numpy>=1.19.0
torch>=2.0.0
torchvision>=0.15.0
lightning>=2.6.0
scipy>=1.7.0
scikit-learn>=1.0.0
Pillow>=8.2.0
h5py>=2.10.0
requests
tqdm>=4.60.0

[docs]
mkdocs==1.1.2
mkdocstrings==0.14.0
mkdocs-material==6.2.5
mkdocs-git-revision-date-localized-plugin==0.8

[extra]
GitPython>=3.0.0
imageio>=2.9.0
moviepy>=1.0.3
psutil>=5.8.0
matplotlib>=3
wandb>=0.10.32
omegaconf>=2.2
hydra-core<1.4,>=1.3
hydra-colorlog>=1.2
hydra-submitit-launcher>=1.2
torch_optimizer>=0.3.0
torchsort>=0.1.4
numba>=0.50.0

[test]
disent[extra]
pytest>=6.2.4
pytest-cov>=2.12.1
