torch
pytorch_lightning
scikit-learn
imbalanced-learn
rdkit
pandas
joblib
h5py
optuna
torchmetrics