Add docs on implementing PyTorch Ops (and CumOp) (#837)

* Add Pytorch support for Cum Op

* Modify test for Cum op

* Raise TypeError if axis not int or None

* Fix init method of CumOp

* Extend tutorial on documentation for Pytorch

* Add tab for Numba

* Add intersphinx mapping

* Parametrize dtype

Author: HarshvirSandhu

doc/conf.py

@@ -32,8 +32,16 @@
     "sphinx.ext.napoleon",
     "sphinx.ext.linkcode",
     "sphinx.ext.mathjax",
+    "sphinx_design",
+    "sphinx.ext.intersphinx"
 ]
 
+intersphinx_mapping = {
+    "jax": ("https://jax.readthedocs.io/en/latest", None),
+    "numpy": ("https://numpy.org/doc/stable", None),
+    "torch": ("https://pytorch.org/docs/stable", None),
+}
+
 needs_sphinx = "3"
 
 todo_include_todos = True

doc/environment.yml

@@ -13,6 +13,7 @@ dependencies:
   - mock
   - pillow
   - pymc-sphinx-theme
+  - sphinx-design
   - pip
   - pip:
     - -e ..