Delete sharded_jit

PiperOrigin-RevId: 464081692

jax/BUILD

@@ -26,7 +26,6 @@ load(
     "py_library_providing_imports_info",
     "pytype_library",
     "scipy_py_deps",
-    "sharded_jit_visibility",
 )
 
 licenses(["notice"])
@@ -97,7 +96,6 @@ py_library_providing_imports_info(
             "_src/test_util.py",
             "*_test.py",
             "**/*_test.py",
-            "interpreters/sharded_jit.py",
         ],
     ) + [
         # until new parallelism APIs are moved out of experimental
@@ -139,7 +137,6 @@ py_library_providing_imports_info(
     visibility = ["//visibility:public"],
     deps = [
         ":jax",
-        ":sharded_jit",
     ] + absl_logging_py_deps + numpy_py_deps,
 )
 
@@ -162,16 +159,6 @@ pytype_library(
     deps = [":jax"],
 )
 
-# sharded_jit is deprecated. Please do not add any more projects to the visibility.
-pytype_library(
-    name = "sharded_jit",
-    srcs = ["interpreters/sharded_jit.py"],
-    visibility = [
-        ":internal",
-    ] + sharded_jit_visibility,
-    deps = [":jax"],
-)
-
 pytype_library(
     name = "optimizers",
     srcs = [

jax/experimental/__init__.py

@@ -12,10 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from jax.interpreters.sharded_jit import (
-  sharded_jit as sharded_jit,
-  with_sharding_constraint as with_sharding_constraint,
-)
 from jax.interpreters.pxla import PartitionSpec as PartitionSpec
 from jax.experimental.x64_context import (
   enable_x64 as enable_x64,