Export LayerNorm (#70)

facebookresearch · Sep 28, 2023 · 3593eca · 3593eca
1 parent 112727d
commit 3593eca
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/fairseq2/nn/__init__.py b/src/fairseq2/nn/__init__.py
@@ -8,6 +8,9 @@
 from fairseq2.nn.incremental_state import IncrementalState as IncrementalState
 from fairseq2.nn.incremental_state import IncrementalStateBag as IncrementalStateBag
 from fairseq2.nn.module_list import ModuleList as ModuleList
+from fairseq2.nn.normalization import LayerNorm as LayerNorm
+from fairseq2.nn.normalization import RMSNorm as RMSNorm
+from fairseq2.nn.normalization import StandardLayerNorm as StandardLayerNorm
 from fairseq2.nn.position_encoder import (
     LearnedPositionEncoder as LearnedPositionEncoder,
 )