patch jax