Have you found any work around for this? I’m working with GPs and for numerical stability reasons the consensus is you need the cholesky_solve …which isn’t supported for .backward calls as you’ve pointed out.