ENH: New function lazy_apply

Author: crusaderky

docs/api-reference.md

@@ -13,6 +13,7 @@
     expand_dims
     isclose
     kron
+    lazy_apply
     nunique
     pad
     setdiff1d

docs/conf.py

@@ -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),
 }
 

pixi.lock

@@ -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" }

pyproject.toml

@@ -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",