diff --git a/docs/source/api/codegen.md b/docs/source/api/codegen.md
index 8e374d4e5a52ba3dd44f338f17c10a2d9fb9c784..c52f28f51944d7d1c7eb35dc1108906c90058c68 100644
--- a/docs/source/api/codegen.md
+++ b/docs/source/api/codegen.md
@@ -177,4 +177,5 @@ The following categories with target-specific options are exposed:
   Kernel
   GpuKernel
   Parameter
+  Lambda
 ```
diff --git a/src/pystencils/codegen/__init__.py b/src/pystencils/codegen/__init__.py
index fc1b70ca0a4dafe4ed099024cd2feb05bf61746b..d06c18382bbc21a7bd9a9a41849d586f88215814 100644
--- a/src/pystencils/codegen/__init__.py
+++ b/src/pystencils/codegen/__init__.py
@@ -6,6 +6,7 @@ from .config import (
 from .parameters import Parameter
 from .kernel import Kernel, GpuKernel
 from .driver import create_kernel, get_driver
+from .functions import Lambda
 
 __all__ = [
     "Target",
@@ -14,6 +15,7 @@ __all__ = [
     "Parameter",
     "Kernel",
     "GpuKernel",
+    "Lambda",
     "create_kernel",
     "get_driver",
 ]