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.