distribution.jax_categorical¶
Jax-based distribution representation.
Classes¶
A categorical distribution stored as a jax array. |
|
Sample type for empirical second-order categorical distributions. |
|
A categorical distribution represented by logits. |
|
A categorical distribution represented by unnormalized probabilities. |