search

JaxGaussianProcesses / JaxLinOp

Linear Operators in JAX.
Apache License 2.0
12 stars 0 forks source link

feat: Matrix multiplication between LinearOperators. #4

Open daniel-dodd opened 1 year ago

daniel-dodd commented 1 year ago

Matrix multiplication is currently only defined between a LinearOperator and a jax.numpy.ndarray.

It would be good to extend this to support LinearOperator to LinearOperator matmuls.