bellini.apiΒΆ

Various modules that interface with the high-performance backends. Only uses numpyro as of now, but should be easily extendable to pytorch/pyro

Modules

bellini.api.functional

A high-level interface that provides a backend-agnostic numpy-like API when performing computations on Quantities and Distributions

bellini.api.utils

A set of utility functions that are useful when interacting with a mix of Quantities, Distributions, and other numerical primitives