brainevent.update_dense_on_binary_post_p#
- brainevent.update_dense_on_binary_post_p = <brainevent.XLACustomKernel object>#
Low-level XLA custom-kernel primitive for
update_dense_on_binary_post.This
XLACustomKernelinstance dispatches the dense weight update for post-synaptic binary plasticity operation to registered backends (numba,pallas), using runtime shape/dtype metadata provided by the high-level wrapper.The operation updates synaptic weights based on postsynaptic spike events and presynaptic trace values: for each postsynaptic neuron
jthat fires,weight[:, j] += pre_trace.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_dense_on_binary_post_p.available_backends(platform), and the default backend can be configured withupdate_dense_on_binary_post_p.set_default(platform, backend).See also
update_dense_on_binary_postHigh-level user-facing function wrapper.