Update Jax and copyright