From b849fc63fd13fcf5a5d47a175e2e5041ebf9b7b7 Mon Sep 17 00:00:00 2001
From: Saurav Maheshkar <sauravvmaheshkar@gmail.com>
Date: Fri, 28 Jun 2024 17:29:35 +0530
Subject: [PATCH] docs: update transformations API docs

---
 docs/api/transformations.rst | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/docs/api/transformations.rst b/docs/api/transformations.rst
index 33395c54..3987c96d 100644
--- a/docs/api/transformations.rst
+++ b/docs/api/transformations.rst
@@ -174,6 +174,10 @@ Transformations and states
 .. autoclass:: NonNegativeParamsState
     :members:
 
+.. autofunction:: normalize_by_update_norm
+.. autoclass:: NormalizeByUpdateNormState
+    :members:
+
 .. autofunction:: per_example_global_norm_clip
 .. autofunction:: per_example_layer_norm_clip