adjusting .gitignore

Signed-off-by: Andrew Schilling <aschilling@nvidia.com>

Author: aschilling-nv

.gitignore

@@ -88,7 +88,6 @@ instance/
 
 # Sphinx documentation
 docs/_build
-docs/apidocs
 
 # PyBuilder
 target/

docs/apidocs/index.md

@@ -0,0 +1,11 @@
+# API Reference
+
+NeMo Emerging Optimizers API reference provides comprehensive technical documentation for all modules, classes, and functions.
+
+```{toctree}
+:caption: API Documentation
+:hidden:
+
+utils.md
+orthogonalized-optimizers.md
+```

docs/apidocs/orthogonalized-optimizers.md

@@ -0,0 +1,32 @@
+```{eval-rst}  
+.. role:: hidden
+    :class: hidden-section
+
+emerging_optimizers.orthogonalized_optimizers
+=============================================
+
+.. automodule:: emerging_optimizers.orthogonalized_optimizers
+.. currentmodule:: emerging_optimizers.orthogonalized_optimizers
+
+:hidden:`OrthogonalizedOptimizer`
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: OrthogonalizedOptimizer
+    :members:
+
+:hidden:`Muon`
+~~~~~~~~~~~~~~~
+
+.. autoclass:: Muon
+    :members:
+
+
+:hidden:`Newton-Schulz`
+~~~~~~~~~~~~~~~~~~~~~~~~
+.. automodule:: emerging_optimizers.orthogonalized_optimizers.muon_utils
+.. currentmodule:: emerging_optimizers.orthogonalized_optimizers.muon_utils
+
+.. autofunction:: newton_schulz
+.. autofunction:: newton_schulz_step
+.. autofunction:: newton_schulz_tp
+```

docs/apidocs/utils.md

@@ -0,0 +1,26 @@
+
+```{eval-rst}  
+.. role:: hidden
+    :class: hidden-section
+
+emerging_optimizers.utils
+==========================
+
+.. automodule:: emerging_optimizers.utils
+.. currentmodule:: emerging_optimizers.utils
+
+:hidden:`fp32_matmul_precision`
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+.. autofunction:: fp32_matmul_precision
+
+:hidden:`get_pg_size`
+~~~~~~~~~~~~~~~~~~~~~
+
+.. autofunction:: get_pg_size
+
+:hidden:`get_pg_rank`
+~~~~~~~~~~~~~~~~~~~~~
+
+.. autofunction:: get_pg_rank
+```