Function register_autograd_module 

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

Register autograd module