Add pystencils.autodiff

README.md

@@ -55,6 +55,7 @@ All options:
 - `alltrafos`: pulls in additional dependencies for loop simplification e.g. libisl
 - `bench_db`: functionality to store benchmark result in object databases
 - `interactive`: installs dependencies to work in Jupyter including image I/O, plotting etc.
+- `autodiff`: enable derivation of adjoint kernels and generation of Torch/Tensorflow operations
 - `doc`: packages to build documentation
 
 Options can be combined e.g.

pystencils/__init__.py

@@ -13,6 +13,13 @@ from .simp import AssignmentCollection
 from .slicing import make_slice
 from .sympyextensions import SymbolCreator
 
+try:
+    import pystencils_autodiff
+    autodiff = pystencils_autodiff
+except Exception:
+    pass
+
+
 __all__ = ['Field', 'FieldType', 'fields',
            'TypedSymbol',
            'make_slice',

pystencils/autodiff.py

+"""
+Provides tools for generation of auto-differentiable operations.
+
+See https://github.com/theHamsta/pystencils_autodiff
+
+Installation:
+
+.. code-block:: bash
+    pip install pyconrad-autodiff
+"""
+import os
+
+if 'CI' not in os.environ:
+    raise NotImplementedError('pystencils-autodiff is not installed. Run `pip install pyconrad-autodiff`')

setup.py

@@ -108,6 +108,7 @@ setup(name='pystencils',
           'alltrafos': ['islpy', 'py-cpuinfo'],
           'bench_db': ['blitzdb', 'pymongo', 'pandas'],
           'interactive': ['matplotlib', 'ipy_table', 'imageio', 'jupyter', 'pyevtk'],
+          'autodiff': ['pystencils_autodiff'],
           'doc': ['sphinx', 'sphinx_rtd_theme', 'nbsphinx',
                   'sphinxcontrib-bibtex', 'sphinx_autodoc_typehints', 'pandoc'],
       },