jax

c5a9eba3 - Implement batched cholesky decomposition using LAPACK/Cusolver (#1956)

Implement batched cholesky decomposition using LAPACK/Cusolver (#1956) * Implement batched Cholesky decomposition on CPU and GPU using LAPACK and cuSolver. Adds support for complex batched Cholesky decomposition on both platforms.. Fix concurrency bug in batched cuBlas kernels where a host to device memcpy could take place too early before the device buffer was ready.