jax authors 5c5baccb2e Merge pull request #16072 from axch:stacked-piles ...

PiperOrigin-RevId: 538839592

2023-06-08 11:22:30 -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

Redefine the pile representation from concatenated to stacked-and-padded.

2023-05-19 13:13:15 -07: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

Copybara import of the project:

2023-05-24 10:35:37 -07:00

xla.py

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

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