Jake VanderPlas 4ae3211ea2 jax.disable_jit: ensure while_loop behaves similarly to non-disable_jit version

2025-03-11 09:53:34 -07:00

..

__init__.py

[Jax][Pallas][Mosaic] Implement platform dependent diag, with branch selection driven by constant prop in mosaic lowering.

2024-12-22 00:50:51 -08:00

common.py

Dedup non-ref constants closed in cond branch functions.

2025-03-07 04:01:42 -08:00

conditionals.py

Dedup non-ref constants closed in cond branch functions.

2025-03-07 04:01:42 -08:00

for_loop.py

[better_errors] Make it explicit that debug_info is not None.

2025-02-13 22:07:04 -08:00

loops.py

jax.disable_jit: ensure while_loop behaves similarly to non-disable_jit version

2025-03-11 09:53:34 -07:00

solves.py

[better_errors] Make it explicit that debug_info is not None.

2025-02-13 22:07:04 -08:00