# Tests
numpy
pytest
pytest-cov
pytest-xdist
dill
setuptools
# Test contrib dependencies
scipy
pytorch_fid==0.1.1
tqdm
scikit-learn
matplotlib
tensorboardX
visdom==0.2.3
# temporary fix for
# ImportError: cannot import name 'soft_unicode' from 'markupsafe'
markupsafe==2.0.1
polyaxon
polyaxon-client
wandb
mlflow
neptune-client>=0.16.17
tensorboard
torchvision
pynvml
clearml
scikit-image
py-rouge
nltk
# Examples dependencies
pandas
gymnasium
mkl
