Tools for lazy backendsΒΆ

These additional functions are meant to be used to support compatibility with lazy backends, e.g. Dask or JAX:

lazy_apply

Lazily apply an eager function.

testing.lazy_xp_function

Tag a function to be tested on lazy backends.

testing.patch_lazy_xp_functions

Test lazy execution of functions tagged with lazy_xp_function().