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

mmcky merged 1 commit intomainQuantEcon/lecture-jax:mainfrom

fix-jax-intro-fori-loopQuantEcon/lecture-jax:fix-jax-intro-fori-loopCopy head branch name to clipboard