minor fixes for JS III

fixes for https://python.quantecon.org/mccall_model_with_sep_markov.html

eliminate 'magic numbers' in @partial(jax.jit, static_argnums=(3, 4))

Get rid of extra compiled version def _simulate_cross_section_compiled

cut # The last parameter is now w_bar (scalar) instead of σ (array)