nn docs

Signed-off-by: Francis Williams <francis@fwilliams.info>

Author: fwilliams

fvdb/nn/__init__.py

@@ -2,26 +2,13 @@
 # SPDX-License-Identifier: Apache-2.0
 #
 from .modules import (
-    CELU,
-    ELU,
-    GELU,
-    SELU,
     AvgPool,
     BatchNorm,
-    Dropout,
     GroupNorm,
-    InjectFromGrid,
-    LeakyReLU,
-    Linear,
     MaxPool,
-    ReLU,
-    Sequential,
-    Sigmoid,
-    SiLU,
     SparseConv3d,
     SparseConvTranspose3d,
     SyncBatchNorm,
-    Tanh,
     UpsamplingNearest,
 )
 from .simple_unet import (
@@ -39,20 +26,9 @@
 __all__ = [
     "AvgPool",
     "BatchNorm",
-    "CELU",
-    "Dropout",
-    "ELU",
-    "GELU",
     "GroupNorm",
     "InjectFromGrid",
-    "LeakyReLU",
-    "Linear",
     "MaxPool",
-    "ReLU",
-    "SELU",
-    "Sequential",
-    "Sigmoid",
-    "SiLU",
     "SimpleUNet",
     "SimpleUNetBasicBlock",
     "SimpleUNetBottleneck",
@@ -65,6 +41,5 @@
     "SparseConv3d",
     "SparseConvTranspose3d",
     "SyncBatchNorm",
-    "Tanh",
     "UpsamplingNearest",
 ]