[docs] add docs for APIs (#72)

docs/requirements.txt

@@ -3,3 +3,4 @@ sphinx==3.2.1
 sphinx_rtd_theme==0.4.3
 sphinxcontrib-programoutput==0.16
 -e git+https://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
+torch=1.6.0

docs/source/api/nn/pipe.rst

+Pipe
+====
+
+.. autoclass:: fairscale.nn.Pipe

docs/source/api/optim/oss.rst

+OSS
+====
+
+.. autoclass:: fairscale.optim.OSS

docs/source/conf.py

@@ -10,19 +10,19 @@
 # add these directories to sys.path here. If the directory is relative to the
 # documentation root, use os.path.abspath to make it absolute, like shown here.
 #
-# import os
-# import sys
-# sys.path.insert(0, os.path.abspath('.'))
+import os
+import sys
 
+sys.path.insert(0, os.path.abspath("../.."))
 
 # -- Project information -----------------------------------------------------
 
-project = 'fairscale'
-copyright = '2020, Facebook AI Research'
-author = 'Facebook AI Research'
+project = "fairscale"
+copyright = "2020, Facebook AI Research"
+author = "Facebook AI Research"
 
 # The full version, including alpha/beta/rc tags
-release = '0.0.2'
+release = "0.0.2"
 
 
 # -- General configuration ---------------------------------------------------
@@ -31,10 +31,11 @@ release = '0.0.2'
 # extensions coming with Sphinx (named 'sphinx.ext.*') or your custom
 # ones.
 extensions = [
+    "sphinx.ext.autodoc",
 ]
 
 # Add any paths that contain templates here, relative to this directory.
-templates_path = ['_templates']
+templates_path = ["_templates"]
 
 # List of patterns, relative to source directory, that match files and
 # directories to ignore when looking for source files.
@@ -47,9 +48,9 @@ exclude_patterns = []
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #
-html_theme = 'alabaster'
+html_theme = "alabaster"
 
 # Add any paths that contain custom static files (such as style sheets) here,
 # relative to this directory. They are copied after the builtin static files,
 # so a file named "default.css" will overwrite the builtin "default.css".
-html_static_path = ['_static']
+html_static_path = ["_static"]

docs/source/index.rst

@@ -7,9 +7,11 @@ Welcome to fairscale's documentation!
 =====================================
 
 .. toctree::
-   :maxdepth: 2
-   :caption: Contents:
+   :maxdepth: 1
+   :caption: API Reference
 
+   api/optim/oss
+   api/nn/pipe
 
 
 Indices and tables