aggregation
===========

.. automodule:: torchjd.aggregation
   :no-members:

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

.. autoclass:: torchjd.aggregation.Aggregator
    :members: __call__

.. autoclass:: torchjd.aggregation.WeightedAggregator
    :members: __call__

.. autoclass:: torchjd.aggregation.GramianWeightedAggregator
    :members: __call__

.. autoclass:: torchjd.aggregation.Weighting
    :members: __call__

.. autoclass:: torchjd.aggregation.GeneralizedWeighting
    :members: __call__

.. autoclass:: torchjd.aggregation.Stateful
    :members: reset


.. toctree::
    :hidden:
    :maxdepth: 1

    upgrad.rst
    aligned_mtl.rst
    cagrad.rst
    config.rst
    constant.rst
    cr_mogm.rst
    dualproj.rst
    flattening.rst
    graddrop.rst
    gradvac.rst
    imtl_g.rst
    krum.rst
    mean.rst
    mgda.rst
    nash_mtl.rst
    pcgrad.rst
    random.rst
    sum.rst
    trimmed_mean.rst