Martin Reinecke / jax_linop