FIX: jax_intro timeout: use lax.fori_loop instead of Python for loop - #442

Merged

mmcky merged 4 commits into

mainfrom

fix/jax-intro-timeout

Nov 28, 2025

Commits on Nov 27, 2025

Fix jax_intro timeout: use lax.fori_loop instead of Python for loop
mmcky
committed
50de16b
style: use jstac's fori_loop naming conventions
mmcky
committed
afc2d36
style: loop_state -> initial_loop_state
mmcky
committed
9bceb2d

Commits on Nov 28, 2025

Merge branch 'main' into fix/jax-intro-timeout
mmcky
authored
a788262