pub(crate) fn register(module: &Bound<'_, PyModule>) -> PyResult<()>
Register differentiable and stochastic learning bindings.