beanmachine.ppl.inference.proposer.nnc package

Submodules

• beanmachine.ppl.inference.proposer.nnc.utils module

Module contents

beanmachine.ppl.inference.proposer.nnc.nnc_jit(f: Callable[[beanmachine.ppl.inference.proposer.nnc.P], beanmachine.ppl.inference.proposer.nnc.R], static_argnums: Optional[Tuple[int]] = None) → Callable[[beanmachine.ppl.inference.proposer.nnc.P], beanmachine.ppl.inference.proposer.nnc.R]

A helper function that lazily imports the NNC utils, which initialize the compiler and displaying a experimental warning, then invoke the underlying nnc_jit on the function f.