Register ModuleBackwardHookState and the wrapper helpers (_create_module_backward_hook_state, _wrap_module_backward_inputs, _wrap_module_backward_outputs) on a pybind11 module.
Parameters
mpy::module_&Target pybind11 module — typically the
engine extension module loaded via from lucid._C import engine as _C_engine.