I\'m using CoxTimeVaryingFitter from lifelines. Other fitters (e.g. CoxPHFitter) have methods to predict the survival function (e.g. CoxPHFitter.
CoxTimeVaryingFitter
CoxPHFitter
CoxPHFitter.