[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 <[email protected]>

* 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 <[email protected]>
Co-authored-by: Copilot <[email protected]>

Author: 3 people