Add lax.fori_loop example and improve jax_intro#530

jstac merged 1 commit intomainQuantEcon/lecture-python-programming:mainfrom

jx3QuantEcon/lecture-python-programming:jx3Copy head branch name to clipboard