distribution

Distribution subpackage.

Submodules

jax_bernoulli	Jax-based Bernoulli distribution representation.
jax_categorical	Jax-based distribution representation.
jax_dirichlet	Jax-based distribution representation.
jax_gaussian	Jax-based Gaussian distribution representation.
numpy_bernoulli	NumPy-based Bernoulli distribution representation.
numpy_categorical	Numpy-based distribution representation.
numpy_dirichlet	Numpy-based distribution representation.
numpy_gaussian	Numpy-based Gaussian distribution representation.
torch_bernoulli	Torch-based Bernoulli distribution representation.
torch_categorical	Torch-based categorical distribution representation.
torch_dirichlet	Torch implementation of the Dirichlet distribution representation.
torch_gaussian	Torch-based Gaussian distribution representation.
torch_mixture	Torch-based mixture distribution representation.
torch_sparse_log_categorical	Sparse torch categorical distributions represented by grouped logits.