sp_block_serialise

hypercoil.functional.sparse.sp_block_serialise(f: Callable, *, n_blocks: int = 1, retnums: Sequence[int] = (), in_axes: Sequence[int] = (), out_axes: Sequence[int] = (), sp_argnums: Sequence[int] = (0,), sp_retnums: Sequence[int] = (0,), sp_retshapes: Sequence[Sequence[int]] = ()) → Callable

Function block serialisation transformation with a convenience wrapper for handling top-k format sparse data.