Add pmean to lax documentation (#2778)

jax-ml · Apr 21, 2020 · 3f59783 · 3f59783
1 parent b36d1bf
commit 3f59783
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docs/jax.lax.rst b/docs/jax.lax.rst
@@ -161,6 +161,7 @@ Parallelism support is experimental.
     psum
     pmax
     pmin
+    pmean
     ppermute
     pswapaxes
     axis_index