This is similar to https://github.com/llvm/llvm-project/pull/127317
NB_DOMAIN
add_mlir_python_extension
See https://mlir.llvm.org/ for more information.