[aiyagari_jax] aiyagari with endogenous labor supply

[shlff]

As discussed with @jstac and mentioned by @aledinola here:

https://discourse.quantecon.org/t/aiyagari-model-with-endogenous-labor-supply/1014/

we can implement the simplest version of the Aiyagari model with endogenous labor supply as an example or exercise.

Reference:

• https://makotonakajima.github.io/files/note/note_im_aiyagari_ext.pdf
• https://www.sas.upenn.edu/~jesusfv/Lecture_Heterogeneous_II.pdf
• https://sciencespo.hal.science/hal-03596961/document

[shlff]

Hi @aledinola, I shifted the discussion from discourse to here.

You are more than welcome to share thoughts and work on this together if you have time and interest.