Sharad Vikram 4d715753c4 Make sure to DCE read effects ...

PiperOrigin-RevId: 738215055

2025-03-18 18:42:14 -07:00

..

__init__.py

Reapply: move jax.interpreters.ad to jax._src.interpreters.ad

2023-02-02 09:29:05 -08:00

ad.py

[direct-linearize] shmap fixes

2025-03-14 21:38:50 +00:00

batching.py

Change the way that batching.spec_types is updated.

2025-03-15 22:58:44 -07:00

mlir.py

Use "x is y" rather than "id(x) == id(y)".

2025-03-14 08:54:46 -07:00

partial_eval.py

Make sure to DCE read effects

2025-03-18 18:42:14 -07:00

pxla.py

Enable jax.device_put to a sharding with no local devices.

2025-03-17 16:49:46 -07:00

xla.py

Re-land changes from https://github.com/jax-ml/jax/pull/25555

2024-12-18 15:02:54 -08:00