distribution.jax_bernoulli¶
Jax-based Bernoulli distribution representation.
Classes¶
A Bernoulli distribution represented as a categorical dstribution with 2 classes. |
|
Sample type for jax Bernoulli distributins. |
|
A Bernoulli distribution represented by class-1 log-odds. |
|
A Bernoulli distribution represented by the probability of class 1. |