This repository was archived by the owner on Nov 17, 2023. It is now read-only.
This repository was archived by the owner on Nov 17, 2023. It is now read-only.
Improve linear algebra functions #14962
Open
Description
One of my projects needs to do matrix inversion which is not very accurate and convenient to do with Cholesky factorization, so I decided to implement myself and improve the linalg package in the same time. New operators to add:
operator | function | progress |
---|---|---|
inverse | matrix inversion | #14963 |
det | matrix determinant | #15007 |
slogdet | signed log determinant | #15007 |
svd | singular value decomposition | |
pinverse | pseudo-inverse (Moore-Penrose inverse) |
Pytorch use MAGMA in their implementation, I'll try to only use cuBLAS and cuSolver in order not to add dependencies.