[career] Convert to JAX using nested vmap (supersedes #617)#985

jstac merged 1 commit into

mainQuantEcon/lecture-python.myst:mainfrom

career-jax-rewriteQuantEcon/lecture-python.myst:career-jax-rewriteCopy head branch name to clipboard