jax

3f597833 - Add pmean to lax documentation (#2778)