pytensor

Bayesian Modeling and Probabilistic Programming in Python

Distributed differentiable graph computation using PyTensor

Estimating Cox Proportional hazards coefficients for for both breslow and efron ties, using both autograd and directly calculating the gradient and hessian