torch
sentence-transformers
mlflow
transformers
hyperopt
xgboost
pynvml
pydantic
pyspark
pytorch-lightning
pytest
pytest-cov
pytest-mock