The example I see in the documentation is COX regression

https://docs.pymc.io/notebooks/survival_analysis.html

I want to estimate the survival function，like this：

https://lifelines.readthedocs.io/en/latest/Survival%20analysis%20with%20lifelines.html