Hi,
the Cholesky class from the notebook linked above is a (not terribly good because it uses “inverse” instead of a triangular solver on a matrix we know to be triangular) autograd.Function that does the same as potrf does on Tensors. diag and prod should work on Variables.
Note that this only works for positive definite matrices (e.g. covariance matrices).
Best regards
Thomas