diff --git a/flax/experimental/nnx/__init__.py b/flax/experimental/nnx/__init__.py
index 8887c072bc..05c82c9ede 100644
--- a/flax/experimental/nnx/__init__.py
+++ b/flax/experimental/nnx/__init__.py
@@ -49,7 +49,6 @@
 from .nnx.nn.activations import log_sigmoid as log_sigmoid
 from .nnx.nn.activations import log_softmax as log_softmax
 from .nnx.nn.activations import logsumexp as logsumexp
-from .nnx.nn.activations import normalize as normalize
 from .nnx.nn.activations import one_hot as one_hot
 from .nnx.nn.activations import relu as relu
 from .nnx.nn.activations import relu6 as relu6
diff --git a/flax/experimental/nnx/nnx/nn/activations.py b/flax/experimental/nnx/nnx/nn/activations.py
index 55fee03d5a..6eec76e42a 100644
--- a/flax/experimental/nnx/nnx/nn/activations.py
+++ b/flax/experimental/nnx/nnx/nn/activations.py
@@ -25,7 +25,6 @@
   log_sigmoid,
   log_softmax,
   logsumexp,
-  normalize,
   one_hot,
   relu,
   relu6,
@@ -53,7 +52,6 @@
   'log_sigmoid',
   'log_softmax',
   'logsumexp',
-  'normalize',
   'one_hot',
   'relu',
   'relu6',