numpy<1.24,>=1.16
torch>=1.4.0
packaging
tqdm
scikit-learn
matplotlib
scipy
pandas
opencv-python
numba
rdkit
ray

[dev]
flake8
pytest
pytest-cov
mypy
