matplotlib
numpy
pandas
lightning
pyyaml
scikit-optimize
scipy
seaborn
torch
torchvision
tqdm
umap-learn
rich
captum
ipywidgets
torch_geometric
ipykernel
lifelines
papermill
scikit-survival
python-louvain
xgboost
pot
geomloss
plotnine
safetensors

[test]
pytest
