[markov_asset] Update to JAX and runtime comparison#654

xuanguang-li wants to merge 14 commits intomainQuantEcon/lecture-python.myst:mainfrom

update_markov_assetQuantEcon/lecture-python.myst:update_markov_assetCopy head branch name to clipboard