pub(super) fn register(module: &Bound<'_, PyModule>) -> PyResult<()>
Register surrogate-gradient LIF training with the extension module.