brainevent.csrmv_p

brainevent.csrmv_p = <brainevent.XLACustomKernel object>

Low-level XLA custom-kernel primitive for csrmv.

This XLACustomKernel instance dispatches the CSR sparse matrix-vector multiplication with floating-point weights operation to registered backends (numba, pallas), using runtime shape/dtype metadata provided by the high-level wrapper.

All elements of the input vector contribute to the result, regardless of sign or magnitude, performing standard sparse matrix-vector multiplication with explicit floating-point weights.

Beyond backend dispatch, the primitive stores JAX transformation bindings (JVP, transpose, batching, and call registration) so the operation integrates correctly with jit, vmap, and autodiff.

Available backends can be queried with csrmv_p.available_backends(platform), and the default backend can be configured with csrmv_p.set_default(platform, backend).

See also

csrmv

High-level user-facing function wrapper.