brainevent.update_csr_on_binary_pre_p

brainevent.update_csr_on_binary_pre_p = <brainevent.XLACustomKernel object>

Low-level XLA custom-kernel primitive for update_csr_on_binary_pre.

This XLACustomKernel instance dispatches the CSR weight update for pre-synaptic binary plasticity operation to registered backends (numba, pallas), using runtime shape/dtype metadata provided by the high-level wrapper.

For each presynaptic neuron that fires, updates all outgoing synaptic weights by adding the corresponding postsynaptic trace values, implementing the pre-synaptic component of spike-timing-dependent plasticity (STDP).

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

See also

update_csr_on_binary_pre

High-level user-facing function wrapper.