.. _models:

Models
======

Abstract base classes
---------------------

.. automodule:: pfl.model.base
   :members:

TensorFlow
----------

.. automodule:: pfl.model.tensorflow
   :members:

PyTorch
-------

.. automodule:: pfl.model.pytorch
   :members:


Gaussian mixture model
----------------------

 .. automodule:: pfl.model.gaussian_mixture_model
   :members:

Exponential moving average
--------------------------

.. automodule:: pfl.model.ema
   :members: