[aiyagari] Update lecture with JAX and check for stylesheet compliance (#590)

* update aiyagari with JAX

* update position of pip install

* minor updates

* Update lectures/aiyagari.md

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

* update timer and install

* update to block until ready

* update install

* incorprate feedback

* minor update

* minor fix

* remove create_firm

---------

Co-authored-by: Matt McKay <mmcky@users.noreply.github.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: 3 people