JAX and the Wayward