Peter Hawkins 07f052a17a [JAX] Remove jax.interpreters.pxla._create_pmap_sharding_spec. ...

No deprecation period: jax.interpreters.pxla is not covered by compatibility policies.

PiperOrigin-RevId: 553911673

2023-08-04 14:23:44 -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

Change to simpler import for jax.config

2023-04-21 11:51:22 -07:00

batching.py

Rename Piles to Jumbles, to avoid unfortunate Imperial entanglements.

2023-07-14 15:19:49 -04:00

mlir.py

Split sharding_impls into its own Bazel target.

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

partial_eval.py

[shape_poly] More cleanup for the internal APIs for shape polymorphism.

2023-07-13 16:37:53 +03:00

pxla.py

[JAX] Remove jax.interpreters.pxla._create_pmap_sharding_spec.

2023-08-04 14:23:44 -07:00

xla.py

Deprecate jax.interpreters.xla.Buffer, device_put, xla_call_p

2023-05-28 07:15:34 -07:00