Probability Distributions¶
This is a collection of probability distributions that can be used as part of a computation graph.
All methods are differentiable, including the sample()
method via the
reparametrization trick or variations thereof. This means that they may be used
in constructing loss functions that require quantities like (cross)entropy or
KL-divergence.