search

normal-computing / posteriors

Uncertainty quantification with PyTorch
https://normal-computing.github.io/posteriors/
Apache License 2.0
314 stars 12 forks source link

Add support for conjugate gradient solver through callable matrix-vector products #73

Closed KaelanDt closed 5 months ago

KaelanDt commented 5 months ago

Adapted from https://jax.readthedocs.io/en/latest/_autosummary/jax.scipy.sparse.linalg.cg.html

SamDuffield commented 5 months ago

Linking to CG discussion #67