[mix_model] Update lecture to use JAX#773

jstac merged 2 commits into

mainQuantEcon/lecture-python.myst:mainfrom

mix_model_fixesQuantEcon/lecture-python.myst:mix_model_fixesCopy head branch name to clipboard