A JAX array with shape (3,).
