some doc fix

Author: refraction-ray

docs/source/conf.py

@@ -21,7 +21,7 @@
 # -- Project information -----------------------------------------------------
 
 project = "tensorcircuit"
-copyright = "2020, refraction-ray"
+copyright = "2020, The TensorCircuit Authors"
 author = "refraction-ray"
 
 # The short X.Y version
@@ -86,13 +86,14 @@
 # The theme to use for HTML and HTML Help pages.  See the documentation for
 # a list of builtin themes.
 #
-html_theme = "alabaster"
+html_theme = "nature"
 
 # Theme options are theme-specific and customize the look and feel of a theme
 # further.  For a list of options available for each theme, see the
 # documentation.
 #
 # html_theme_options = {}
+# currently nature theme doesn't support these github related options (refraction-ray)
 html_theme_options = {
     "github_user": "refraction-ray",
     "github_repo": "tensorcircuit",

docs/source/contribution.rst

@@ -26,7 +26,7 @@ The following git workflow is recommended to contribute by PR:
 
 * Fork the tensorcircuit GitHub repository by clicking the Fork button from GitHub. This will create a fork version of the code repository in your own GitHub account.
 
-* Configure the python enviroment locally for development. ``pip install -r requirements.txt`` and ``pip install -r requirements-dev.txt`` are recommended. Extra packages may be required for specific development tasks.
+* Configure the python environment locally for development. ``pip install -r requirements.txt`` and ``pip install -r requirements-dev.txt`` are recommended. Extra packages may be required for specific development tasks.
 
 * Clone your fork repository locally and setup upstreams to the official tensorcircuit repository. And configure your git user and email so that they match your GitHub account if you haven't.
 

tensorcircuit/backends/jax_backend.py

@@ -117,6 +117,11 @@ def _svd_jax(
 
 
 class JaxBackend(jax_backend.JaxBackend):  # type: ignore
+    """
+    see the original backend API at `jax backend
+    <https://github.com/google/TensorNetwork/blob/master/tensornetwork/backends/jax/jax_backend.py>`_
+    """
+
     # Jax doesn't support 64bit dtype, unless claim
     # ``from jax.config import config```
     # ``config.update("jax_enable_x64", True)``

tensorcircuit/backends/numpy_backend.py

@@ -41,6 +41,11 @@ def _sum_numpy(
 
 
 class NumpyBackend(numpy_backend.NumPyBackend):  # type: ignore
+    """
+    see the original backend API at `numpy backend
+    <https://github.com/google/TensorNetwork/blob/master/tensornetwork/backends/numpy/numpy_backend.py>`_
+    """
+
     def eye(
         self, N: int, dtype: Optional[str] = None, M: Optional[int] = None
     ) -> Tensor:

tensorcircuit/backends/pytorch_backend.py

@@ -53,6 +53,14 @@ def _sum_torch(
 
 
 class PyTorchBackend(pytorch_backend.PyTorchBackend):  # type: ignore
+    """
+    see the original backend API at `pytorch backend
+    <https://github.com/google/TensorNetwork/blob/master/tensornetwork/backends/pytorch/pytorch_backend.py>`_
+
+    Note the functionality provided by pytorch backend is incomplete,
+    it currenly lacks native efficicent jit and vmap support.
+    """
+
     def __init__(self) -> None:
         super(PyTorchBackend, self).__init__()
         global torchlib

tensorcircuit/backends/tensorflow_backend.py

@@ -121,6 +121,11 @@ def _random_choice_tf(
 
 
 class TensorFlowBackend(tensorflow_backend.TensorFlowBackend):  # type: ignore
+    """
+    see the original backend API at `tensorflow backend
+    <https://github.com/google/TensorNetwork/blob/master/tensornetwork/backends/tensorflow/tensorflow_backend.py>`_
+    """
+
     def __init__(self) -> None:
         global tf
         super(TensorFlowBackend, self).__init__()

tensorcircuit/keras.py

@@ -133,7 +133,7 @@ def save_func(f: Callable[..., Any], path: str) -> None:
     """
     save tf function in file (``tf.savedmodel`` format)
 
-    :param f: ``tf.function``ed function with graph building
+    :param f: ``tf.function`` ed function with graph building
     :type f: Callable[..., Any]
     :param path: the dir path to save the function
     :type path: str