Michael Levesque-Dion 43f51d73ce Clean up version switches from dense array migration ...

PiperOrigin-RevId: 637955865

2024-05-28 10:58:51 -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

Finalize deprecations of jax.interpreters.ad config & source_info_util

2024-03-18 13:33:17 -07:00

batching.py

del add_any_p and zeros_like_p, replace aval-dispatched traceable

2023-12-21 17:04:21 -08:00

mlir.py

Clean up version switches from dense array migration

2024-05-28 10:58:51 -07:00

partial_eval.py

integrate attrs in jax.jit

2024-01-27 17:44:43 -08:00

pxla.py

Delete mesh_sharding_specs from JAX

2024-01-02 11:14:39 -08:00

xla.py

Removed noop # type: ignore comments

2024-05-19 21:01:29 +01:00