docs: update transformations

google-deepmind · Jun 28, 2024 · 5c01f6f · 5c01f6f
1 parent c2a7714
commit 5c01f6f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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