fix up extend:core build rule

We want `pytype_strict_library` here.

PiperOrigin-RevId: 624337356

Author: froystig

jax/extend/BUILD

@@ -38,6 +38,7 @@ pytype_strict_library(
 py_library_providing_imports_info(
     name = "core",
     srcs = glob(["core/**/*.py"]),
+    lib_rule = pytype_strict_library,
     deps = [
         "//jax",
         "//jax:abstract_arrays",