Function register_optim_module 

pub fn register_optim_module(
    _py: Python<'_>,
    m: &Bound<'_, PyModule>,
) -> PyResult<()>

Register the optim module with Python