Peter Hawkins 54ff78dbde Deprecate jax.interpreters.xla.Device and jax.interpreters.xla.DeviceArray. ...

PiperOrigin-RevId: 508502470

2023-02-09 16:11:48 -08: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

move jax.interpreters.batching to jax._src.interpreters.batching

2023-02-08 09:51:00 -08:00

mlir.py

Move jax.interpreters.mlir to jax._src.interpreters.mlir.

2023-02-08 14:39:01 -08:00

partial_eval.py

Remove jax_experimental_subjaxpr_lowering_cache since it was only for jit and was False by default. Now that jit/pjit are merged, this cache is not needed since pjit does the caching and we get it for free.

2023-02-08 14:55:56 -08:00

pxla.py

Change jax.ad, jax.xla, jax.pxla to point to the shims instead of the internal modules.

2023-02-09 13:31:40 -08:00

xla.py

Deprecate jax.interpreters.xla.Device and jax.interpreters.xla.DeviceArray.

2023-02-09 16:11:48 -08:00