doc

e3nn · Oct 9, 2023 · 74ee5ac · 74ee5ac
1 parent a771d08
commit 74ee5ac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/docs/api/extra.rst b/docs/api/extra.rst
@@ -11,4 +11,10 @@ Extra Stuff
 .. autofunction:: e3nn_jax.scatter_sum
 
 
+.. autofunction:: e3nn_jax.scatter_mean
+
+
+.. autofunction:: e3nn_jax.scatter_max
+
+
 .. autofunction:: e3nn_jax.radius_graph
diff --git a/docs/api/flax.rst b/docs/api/flax.rst
@@ -7,3 +7,7 @@ Flax Modules
 
 .. autoclass:: e3nn_jax.flax.MultiLayerPerceptron
     :members:
+
+
+.. autoclass:: e3nn_jax.flax.BatchNorm
+    :members: