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.