Computation

MultiOutputKernelComputation

Bases: AbstractKernelComputation

Compute engine for multi-output kernels.

Iterates over kernel.components — a sequence of (CoregionalizationMatrix, kernel) pairs — to build structured covariance matrices. Single-component kernels (ICM) retain Kronecker structure; multi-component kernels (LCM) materialise the sum to Dense.

cross_covariance

cross_covariance(
    kernel, x: Num[Array, "N D"], y: Num[Array, "M D"]
) -> Float[Array, ...]

Override to bypass [N, M] return type annotation for multi-output.