Open ph-frank opened 2 months ago
Introduces custom jvps for matrix inverse and matrix square root to enable stabilize autodiff rules for GP matrices in ICR (Requires eigendecomposition).
Followed up in https://gitlab.mpcdf.mpg.de/ift/nifty/-/merge_requests/972 .
Introduces custom jvps for matrix inverse and matrix square root to enable stabilize autodiff rules for GP matrices in ICR (Requires eigendecomposition).