brainevent.jits_p

brainevent.jits_p = <brainevent.XLACustomKernel object>

Low-level XLA custom-kernel primitive for jits.

This XLACustomKernel instance dispatches the JIT scalar connectivity matrix generation operation to registered backends (numba, pallas), using runtime shape/dtype metadata provided by the high-level wrapper.

This operation generates a sparse connectivity matrix where all non-zero weights are set to the same scalar value. The connectivity pattern is generated on-the-fly using a deterministic PRNG seeded by the provided seed value.

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 jits_p.available_backends(platform), and the default backend can be configured with jits_p.set_default(platform, backend).

See also

jits

High-level user-facing function wrapper.