Compute the Cholesky decomposition of a (batched) positive-definite matrix.
This is like jax.lax.linalg.cholesky(), except with an option to symmetrize the input matrix, which is on by default.
jax.lax.linalg.cholesky()
Compute the Cholesky decomposition of a (batched) positive-definite matrix.
This is like
jax.lax.linalg.cholesky(), except with an option to symmetrize the input matrix, which is on by default.