Update Jax for IE 11 support