I got this warning when calling jnp.cholesky with the torch backend:
UserWarning: torch.cholesky is deprecated in favor of torch.linalg.cholesky and will be removed in a future PyTorch release.
L = torch.cholesky(A)
should be replaced with
L = torch.linalg.cholesky(A)
and
U = torch.cholesky(A, upper=True)
should be replaced with
U = torch.linalg.cholesky(A).mH
This transform will produce equivalent results for all valid (symmetric positive definite) inputs. (Triggered internally at third_party/py/torch/aten/src/ATen/native/BatchLinearAlgebra.cpp:1703.)
return torch.cholesky(x)
I got this warning when calling jnp.cholesky with the torch backend: