George Necula 817b3e5757 [better_errors] Continue adding debug info to Jaxprs (step 7) ...

This follows in a series, starting with #26078 and #26313, adding debug_info to more calls to lu.wrap_init.

Fixes in jet, stateful code, key_reuse, ode, pallas, tests.

2025-02-09 18:14:33 +02:00

..

__init__.py

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

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

ad.py

Merge pull request #26348 from gnecula:debug_info_jaxpr_3

2025-02-06 06:59:18 -08:00

batching.py

[better_errors] Continue adding debug info to Jaxprs (step 3)

2025-02-06 16:26:49 +02:00

mlir.py

Merge pull request #26313 from gnecula:debug_info_vjp

2025-02-05 10:58:10 -08:00

partial_eval.py

tweak an error message on mutable arrays

2025-02-06 17:16:32 +00:00

pxla.py

[better_errors] Continue adding debug info to Jaxprs (step 7)

2025-02-09 18:14:33 +02:00

xla.py

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

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