data-apis
diff --git a/‎docs/api-reference.md
+1 b/‎docs/api-reference.md
+1
diff --git a/‎docs/conf.py
+2 b/‎docs/conf.py
+2
diff --git a/‎pixi.lock
+30-1 b/‎pixi.lock
+30-1
diff --git a/‎pyproject.toml
+3 b/‎pyproject.toml
+3
diff --git a/‎src/array_api_extra/__init__.py
+2 b/‎src/array_api_extra/__init__.py
+2
@@ -13,6 +13,7 @@
     expand_dims
     isclose
     kron
+    lazy_apply
     nunique
     pad
     setdiff1d
 
@@ -53,6 +53,8 @@
 
 intersphinx_mapping = {
     "python": ("https://docs.python.org/3", None),
+    "numpy": ("https://numpy.org/doc/stable", None),
+    "dask": ("https://docs.dask.org/en/stable", None),
     "jax": ("https://jax.readthedocs.io/en/latest", None),
 }
 
 
@@ -98,6 +98,7 @@ furo = ">=2023.08.17"
 myst-parser = ">=0.13"
 sphinx-copybutton = "*"
 sphinx-autodoc-typehints = "*"
+numpy = "*"
 
 [tool.pixi.feature.docs.tasks]
 docs = { cmd = "sphinx-build . build/", cwd = "docs" }
@@ -283,6 +284,8 @@ messages_control.disable = [
     "fixme",                      # allow FIXME comments
     "line-too-long",              # ruff handles this
     "missing-function-docstring", # numpydoc handles this
+    "import-error",               # mypy handles this
+    "import-outside-toplevel",    # optional dependencies
 ]
 
 
 
@@ -12,6 +12,7 @@
     setdiff1d,
     sinc,
 )
+from ._lib._lazy import lazy_apply
 
 __version__ = "0.6.1.dev0"
 
@@ -25,6 +26,7 @@
     "expand_dims",
     "isclose",
     "kron",
+    "lazy_apply",
     "nunique",
     "pad",
     "setdiff1d",
Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,8 @@`
`53`	`53`
`54`	`54`	`intersphinx_mapping = {`
`55`	`55`	`"python": ("https://docs.python.org/3", None),`
	`56`	`+ "numpy": ("https://numpy.org/doc/stable", None),`
	`57`	`+ "dask": ("https://docs.dask.org/en/stable", None),`
`56`	`58`	`"jax": ("https://jax.readthedocs.io/en/latest", None),`
`57`	`59`	`}`
`58`	`60`