index.rst

aggregation

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

Abstract base classes

.. autoclass:: torchjd.aggregation.Aggregator
    :members:
    :undoc-members:
    :exclude-members: forward

.. autoclass:: torchjd.aggregation.Weighting
    :members:
    :undoc-members:
    :exclude-members: forward

.. autoclass:: torchjd.aggregation.GeneralizedWeighting
    :members:
    :undoc-members:
    :exclude-members: forward

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

    upgrad.rst
    aligned_mtl.rst
    cagrad.rst
    config.rst
    constant.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