torch
numpy
pandas
scikit-learn
optuna
lifetimes
matplotlib

[dev]
pytest

[wandb]
wandb
