diff --git a/docs/src/python/nn/functions.rst b/docs/src/python/nn/functions.rst
index 9b6cd9f62..325a24ef2 100644
--- a/docs/src/python/nn/functions.rst
+++ b/docs/src/python/nn/functions.rst
@@ -28,6 +28,7 @@ simple functions.
    prelu
    relu
    relu6
+   relu_squared
    selu
    sigmoid
    silu
diff --git a/docs/src/python/nn/layers.rst b/docs/src/python/nn/layers.rst
index 4eb14b088..218f7f457 100644
--- a/docs/src/python/nn/layers.rst
+++ b/docs/src/python/nn/layers.rst
@@ -51,6 +51,7 @@ Layers
    RMSNorm
    ReLU
    ReLU6
+   ReLUSquared
    RNN
    RoPE
    SELU