Matthew Johnson b05975b964 add result info to mhlo, fixes #14780 ...

incidentally fixes #14787

2023-03-06 21:21:26 -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

prune unintended exports from jax.interpreters.batching

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

mlir.py

Reexport jax.interpreters.mlir.token_type.

2023-02-17 13:26:44 +00:00

partial_eval.py

add result info to mhlo, fixes #14780

2023-03-06 21:21:26 -08:00

pxla.py

Internal change

2023-03-04 13:24:11 +00:00

xla.py

[JAX] Move jax._src.lib.xla_bridge to jax._src.xla_bridge.

2023-02-28 07:01:57 -08:00