[inventory_dynamics] Update code to JAX and latest style guide#623

kp992 wants to merge 8 commits intomainQuantEcon/lecture-python.myst:mainfrom

inventory_dynsQuantEcon/lecture-python.myst:inventory_dynsCopy head branch name to clipboard