diff --git a/src/pystencils_autodiff/__init__.py b/src/pystencils_autodiff/__init__.py index bb4cb1845ab8e64c938fc01af02f2a0eec20d53e..c90c5b8106ff5d924ac07ee13fa4b7bb9e16caa7 100644 --- a/src/pystencils_autodiff/__init__.py +++ b/src/pystencils_autodiff/__init__.py @@ -17,9 +17,10 @@ __all__ = ['backends', 'create_backward_assignments', 'AutoDiffOp', 'AutoDiffAstPair', - "tf_constant_from_field", " tf_placeholder_from_field", - "tf_scalar_variable_from_field", " tf_variable_from_field", - "torch_tensor_from_field", "DiffModes"] + 'tensorflow_jit', + 'tf_constant_from_field', ' tf_placeholder_from_field', + 'tf_scalar_variable_from_field', ' tf_variable_from_field', + 'torch_tensor_from_field', 'DiffModes'] sys.modules['pystencils.autodiff'] = pystencils_autodiff sys.modules['pystencils.autodiff.backends'] = pystencils_autodiff.backends