jax

93e11433 - improve disable_jit docstring