Peter Hawkins 33acdc0e40 Delete some dead code that pertained to sharded_jit. ...

sharded_jit is long gone.

PiperOrigin-RevId: 523711890

2023-04-12 08:49:54 -07:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

ad.py

Prune accidentally exported names from jax.interpreters.ad.

2023-02-06 14:36:44 -08:00

batching.py

prune unintended exports from jax.interpreters.batching

2023-02-10 16:47:28 -08:00

mlir.py

Split sharding_impls into its own Bazel target.

2023-04-10 10:15:58 -07:00

partial_eval.py

Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.

2023-03-27 13:30:47 -07:00

pxla.py

Delete some dead code that pertained to sharded_jit.

2023-04-12 08:49:54 -07:00

xla.py

Split sharding_impls into its own Bazel target.

2023-04-10 10:15:58 -07:00