brainevent.update_csr_on_binary_post_p

brainevent.update_csr_on_binary_post_p = <brainevent.XLACustomKernel object>

Low-level XLA custom-kernel primitive for update_csr_on_binary_post.

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

For each postsynaptic neuron that fires, updates all incoming synaptic weights by adding the corresponding presynaptic trace values, implementing the post-synaptic component of spike-timing-dependent plasticity (STDP). Uses a CSC structure internally for efficient iteration over postsynaptic spikes.

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

See also

update_csr_on_binary_post

High-level user-facing function wrapper.