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