Migrate StableHLO Python extension to nanobind.

I'm working towards moving the MLIR Python core code to use nanobind instead of pybind11:
* llvm/llvm-project#117922, which was merged recently, allows downstream Python dialect extensions to be defined using either pybind11 or nanobind.
* llvm/llvm-project#118583 is a PR in review that ports the Python core code to use nanobind instead of pybind11.

This PR migrates StableHLO and related dialects to use nanobind rather than pybind11, with the goal of migrating JAX away from pybind11.

PiperOrigin-RevId: 705197067

Author: hawkinsp

tensorflow/compiler/mlir/stablehlo/BUILD

@@ -38,12 +38,12 @@ tsl_pybind_extension(
         "//tensorflow/tools/pip_package:__subpackages__",
     ],
     deps = [
+        "//third_party/nanobind",
         "//third_party/python_runtime:headers",
         "@llvm-project//llvm:Support",
         "@llvm-project//mlir:CAPIIR",
         "@llvm-project//mlir:IR",
         "@llvm-project//mlir:MLIRBindingsPythonHeadersAndDeps",
-        "@pybind11",
         "@stablehlo//:stablehlo_capi",
     ],
 )

tensorflow/compiler/mlir/stablehlo/stablehlo.cc

@@ -13,13 +13,13 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
-#include "pybind11/pybind11.h"  // from @pybind11
+#include "third_party/nanobind/include/nanobind/nanobind.h"
 #include "stablehlo/integrations/python/StablehloApi.h"  // from @stablehlo
 
 namespace mlir {
 namespace stablehlo {
 
-PYBIND11_MODULE(stablehlo_extension, m) { mlir::stablehlo::AddPortableApi(m); }
+NB_MODULE(stablehlo_extension, m) { mlir::stablehlo::AddPortableApi(m); }
 
 }  // namespace stablehlo
 }  // namespace mlir