Distribution
============

.. automodule:: pfl.internal.distribution.distribution
   :members:

Diagonal Gaussian
-----------------

.. automodule:: pfl.internal.distribution.diagonal_gaussian
   :members:

Log float
---------

.. automodule:: pfl.internal.distribution.log_float
   :members:

Log float functions
-------------------

.. automodule:: pfl.internal.distribution.log_float_functions
   :members:

Mixture
-------

.. automodule:: pfl.internal.distribution.mixture
   :members: