Thanks for this great library!
I was wondering if you plan to include censored loss functions as well to allow for censored probabilistic regression. Similar to what XGB AFT models do, but with a non-constant scale parameters in the distribution models (https://xgboost.readthedocs.io/en/stable/tutorials/aft_survival_analysis.html).