Hide accidental exports from jax.core.

PiperOrigin-RevId: 511350939

jax/core.py

@@ -32,9 +32,6 @@
   Effect as Effect,
   Effects as Effects,
   EvalTrace as EvalTrace,
-  FLAGS as FLAGS,
-  HashableFunction as HashableFunction,
-  HashableWrapper as HashableWrapper,
   InDBIdx as InDBIdx,
   InconclusiveDimensionOperation as InconclusiveDimensionOperation,
   InputType as InputType,
@@ -73,7 +70,6 @@
   apply_todos as apply_todos,
   as_hashable_function as as_hashable_function,
   as_named_shape as as_named_shape,
-  attrgetter as attrgetter,
   aval_mapping_handlers as aval_mapping_handlers,
   axis_frame as axis_frame,
   call as call,
@@ -91,12 +87,8 @@
   concrete_aval as concrete_aval,
   concrete_or_error as concrete_or_error,
   concretization_function_error as concretization_function_error,
-  config as config,
-  contextmanager as contextmanager,
   cur_sublevel as cur_sublevel,
-  curry as curry,
   custom_typechecks as custom_typechecks,
-  dataclass as dataclass,
   dedup_referents as dedup_referents,
   diff_dim as diff_dim,
   diff_shape as diff_shape,
@@ -114,24 +106,18 @@
   extend_axis_env_nd as extend_axis_env_nd,
   find_top_trace as find_top_trace,
   full_lower as full_lower,
-  functools as functools,
-  gc as gc,
   gensym as gensym,
   get_aval as get_aval,
   get_referent as get_referent,
   greater_equal_dim as greater_equal_dim,
   greater_equal_shape as greater_equal_shape,
   has_opaque_dtype as has_opaque_dtype,
-  inspect as inspect,
   is_constant_dim as is_constant_dim,
   is_constant_shape as is_constant_shape,
   is_dim as is_dim,
   is_empty_shape as is_empty_shape,
   is_opaque_dtype as is_opaque_dtype,
   is_special_dim_size as is_special_dim_size,
-  it as it,
-  jax_config as jax_config,
-  jax_jit as jax_jit,
   jaxpr_as_fun as jaxpr_as_fun,
   jaxpr_uses_outfeed as jaxpr_uses_outfeed,
   jaxprs_in_params as jaxprs_in_params,
@@ -153,12 +139,9 @@
   new_sublevel as new_sublevel,
   no_axis_name as no_axis_name,
   no_effects as no_effects,
-  np as np,
   opaque_dtypes as opaque_dtypes,
-  operator as operator,
   outfeed_primitives as outfeed_primitives,
   partial as partial,
-  partialmethod as partialmethod,
   pp as pp,
   pp_aval as pp_aval,
   pp_eqn as pp_eqn,
@@ -176,7 +159,6 @@
   primitive_uses_outfeed as primitive_uses_outfeed,
   process_env_traces_call as process_env_traces_call,
   process_env_traces_map as process_env_traces_map,
-  prod as prod,
   pytype_aval_mappings as pytype_aval_mappings,
   raise_as_much_as_possible as raise_as_much_as_possible,
   raise_to_shaped as raise_to_shaped,
@@ -204,7 +186,6 @@
   symbolic_equal_one_of_dim as symbolic_equal_one_of_dim,
   symbolic_equal_shape as symbolic_equal_shape,
   thread_local_state as thread_local_state,
-  threading as threading,
   token as token,
   total_ordering as total_ordering,
   trace_state_clean as trace_state_clean,
@@ -215,17 +196,9 @@
   typecheck as typecheck,
   typecompat as typecompat,
   typematch as typematch,
-  types as types,
-  typing as typing,
   unmapped_aval as unmapped_aval,
-  unsafe_map as unsafe_map,
-  unsafe_zip as unsafe_zip,
   used_axis_names as used_axis_names,
   used_axis_names_jaxpr as used_axis_names_jaxpr,
   valid_jaxtype as valid_jaxtype,
-  warnings as warnings,
-  weakref_lru_cache as weakref_lru_cache,
   zip as zip,
 )
-
-from typing import Any, Callable, Dict, Type