Yash Katariya 5192fca09b Delete mesh_sharding_specs from JAX ...

PiperOrigin-RevId: 595164505

2024-01-02 11:14:39 -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

revise logic for tangent types of extended dtypes

2023-12-20 14:24:52 -08: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

Integrate StableHLO at openxla/stablehlo@ab709fe4

2023-12-11 12:30:50 -08: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

Delete mesh_sharding_specs from JAX

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

xla.py

Remove a number of deprecated APIs

2023-12-18 10:08:47 -08:00