pub(crate) fn register(m: &Bound<'_, PyModule>) -> PyResult<()>
Register the compute-graph IR classes and functions on the Python extension module.