JAX: Difference between revisions