tensorflow>2.0
scikit-learn
tqdm
pytest
tensorflow_probability
