From 5c01f6f861e518f5d829c640061d26cfb1318659 Mon Sep 17 00:00:00 2001
From: Saurav Maheshkar <sauravvmaheshkar@gmail.com>
Date: Fri, 28 Jun 2024 14:47:43 +0530
Subject: [PATCH] docs: update transformations

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

diff --git a/docs/api/transformations.rst b/docs/api/transformations.rst
index c1fcb9cc..33395c54 100644
--- a/docs/api/transformations.rst
+++ b/docs/api/transformations.rst
@@ -32,6 +32,8 @@ Transformations
     identity
     keep_params_nonnegative
     NonNegativeParamsState
+    normalize_by_update_norm
+    NormalizeByUpdateNormState
     OptState
     Params
     per_example_global_norm_clip