Module: tfp.experimental.fastgp.mbcg

Modified Batched Conjugate Gradients (mBCG) for JAX.

View aliases

Main aliases

`tfp.experimental.fastgp.fast_gp.fast_log_det.mbcg`, `tfp.experimental.fastgp.fast_gp.fast_log_det.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_gp.fast_log_det.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_gp.mbcg`, `tfp.experimental.fastgp.fast_gp.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_log_det.mbcg`, `tfp.experimental.fastgp.fast_log_det.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_log_det.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.fast_log_det.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.fast_log_det.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.fast_log_det.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_gp.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_log_det.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_log_det.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_mtgp.fast_log_det.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.fast_mtgp.preconditioners.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.linalg.partial_lanczos.mbcg`, `tfp.experimental.fastgp.partial_lanczos.mbcg`, `tfp.experimental.fastgp.preconditioners.linalg.partial_lanczos.mbcg`

Classes

class Array: Array base class for JAX

class SymmetricTridiagonalMatrix: Holds a batch of symmetric, tridiagonal matrices.

Functions

modified_batched_conjugate_gradients(...): Return A^(-1)B and Lanczos tridiagonal matrices.

safe_stack(...): Like jnp.stack, but handles len == 0 or 1.

tridiagonal_det(...): Return the determinant of a tridiagonal matrix.