Typing

FunctionalSample module-attribute

FunctionalSample = Callable[
    [Float[Array, "N D"]], Float[Array, "N B"]
]

Type alias for functions representing $B$ samples from a model, to be evaluated on any set of $N$ inputs (of dimension $D$) and returning the evaluations of each (potentially approximate) sample draw across these inputs.