jax authors 1703f096b5 Merge pull request #15088 from froystig:custom-vjp-symbolic-zeros ...

PiperOrigin-RevId: 518433198

2023-03-21 18:05:33 -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

Reexport jax.interpreters.mlir.token_type.

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

partial_eval.py

custom_vjp symbolic zeros support

2023-03-21 14:14:35 +00:00

pxla.py

Mark jax.interpreters.pxla.ShardedDeviceArray as deprecated.

2023-03-21 05:13:55 -07:00

xla.py

[Jax cleanup]

2023-03-16 15:47:28 -07:00