A JAX array with shape (12,) and dtype float32.