Custom Operators

Custom Lineax operators for GPJax.

BlockDiag

BlockDiag(blocks)

Bases: AbstractLinearOperator

Block diagonal linear operator.

Kronecker

Kronecker(A, B)

Bases: AbstractLinearOperator

Kronecker product linear operator with efficient mv via the vec trick.