data-apis
diff --git a/‎docs/api-lazy.md
+15 b/‎docs/api-lazy.md
+15
diff --git a/‎docs/index.md
+1-1 b/‎docs/index.md
+1-1
diff --git a/‎docs/testing-utils.md
-14 b/‎docs/testing-utils.md
-14
diff --git a/‎pixi.lock
+30-1 b/‎pixi.lock
+30-1
diff --git a/‎pyproject.toml
+1 b/‎pyproject.toml
+1
diff --git a/‎src/array_api_extra/__init__.py
+2 b/‎src/array_api_extra/__init__.py
+2
@@ -0,0 +1,15 @@
+# Tools for lazy backends
+
+These additional functions are meant to be used to support compatibility with
+lazy backends, e.g. Dask or Jax:
+
+```{eval-rst}
+.. currentmodule:: array_api_extra
+.. autosummary::
+    :nosignatures:
+    :toctree: generated
+
+    lazy_apply
+    testing.lazy_xp_function
+    testing.patch_lazy_xp_functions
+```
@@ -5,7 +5,7 @@
 :hidden:
 self
 api-reference.md
-testing-utils.md
+api-lazy.md
 contributing.md
 contributors.md
 ```
 
@@ -106,6 +106,7 @@ sphinx-autodoc-typehints = "*"
 dask-core = "*"
 pytest = "*"
 typing-extensions = "*"
+numpy = "*"
 
 [tool.pixi.feature.docs.tasks]
 docs = { cmd = "sphinx-build . build/", cwd = "docs" }
 
@@ -13,6 +13,7 @@
     setdiff1d,
     sinc,
 )
+from ._lib._lazy import lazy_apply
 
 __version__ = "0.7.0.dev0"
 
@@ -27,6 +28,7 @@
     "expand_dims",
     "isclose",
     "kron",
+    "lazy_apply",
     "nunique",
     "pad",
     "setdiff1d",