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

[test]
pytest
