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

mmcky merged 4 commits intomainQuantEcon/lecture-python-programming.myst:mainfrom

fix/jax-intro-timeoutQuantEcon/lecture-python-programming.myst:fix/jax-intro-timeoutCopy head branch name to clipboard