Index

_

__call__() (torchjd.aggregation.Aggregator method) (torchjd.aggregation.AlignedMTL method) (torchjd.aggregation.AlignedMTLWeighting method) (torchjd.aggregation.CAGrad method) (torchjd.aggregation.CAGradWeighting method) (torchjd.aggregation.ConFIG method) (torchjd.aggregation.Constant method) (torchjd.aggregation.ConstantWeighting method) (torchjd.aggregation.DualProj method) (torchjd.aggregation.DualProjWeighting method) (torchjd.aggregation.Flattening method) (torchjd.aggregation.GeneralizedWeighting method) (torchjd.aggregation.GradDrop method) (torchjd.aggregation.GradVac method) (torchjd.aggregation.GradVacWeighting method) (torchjd.aggregation.IMTLG method) (torchjd.aggregation.IMTLGWeighting method) (torchjd.aggregation.Krum method) (torchjd.aggregation.KrumWeighting method) (torchjd.aggregation.Mean method) (torchjd.aggregation.MeanWeighting method) (torchjd.aggregation.MGDA method) (torchjd.aggregation.MGDAWeighting method) (torchjd.aggregation.NashMTL method) (torchjd.aggregation.PCGrad method) (torchjd.aggregation.PCGradWeighting method) (torchjd.aggregation.Random method) (torchjd.aggregation.RandomWeighting method) (torchjd.aggregation.Sum method) (torchjd.aggregation.SumWeighting method) (torchjd.aggregation.TrimmedMean method) (torchjd.aggregation.UPGrad method) (torchjd.aggregation.UPGradWeighting method) (torchjd.aggregation.Weighting method)

A

Aggregator (class in torchjd.aggregation)

AlignedMTL (class in torchjd.aggregation) AlignedMTLWeighting (class in torchjd.aggregation)

B

backward() (in module torchjd.autojac)

C

CAGrad (class in torchjd.aggregation) CAGradWeighting (class in torchjd.aggregation) compute_gramian() (torchjd.autogram.Engine method)

ConFIG (class in torchjd.aggregation) Constant (class in torchjd.aggregation) ConstantWeighting (class in torchjd.aggregation)

D

DualProj (class in torchjd.aggregation)

DualProjWeighting (class in torchjd.aggregation)

E

Engine (class in torchjd.autogram)

F

Flattening (class in torchjd.aggregation)

G

GeneralizedWeighting (class in torchjd.aggregation) GradDrop (class in torchjd.aggregation)

GradVac (class in torchjd.aggregation) GradVacWeighting (class in torchjd.aggregation)

I

IMTLG (class in torchjd.aggregation)

IMTLGWeighting (class in torchjd.aggregation)

J

jac() (in module torchjd.autojac)

jac_to_grad() (in module torchjd.autojac)

K

Krum (class in torchjd.aggregation)

KrumWeighting (class in torchjd.aggregation)

M

Mean (class in torchjd.aggregation) MeanWeighting (class in torchjd.aggregation) MGDA (class in torchjd.aggregation) MGDAWeighting (class in torchjd.aggregation)

module torchjd.aggregation torchjd.autogram torchjd.autojac mtl_backward() (in module torchjd.autojac)

N

NashMTL (class in torchjd.aggregation)

P

PCGrad (class in torchjd.aggregation)

PCGradWeighting (class in torchjd.aggregation)

R

Random (class in torchjd.aggregation) RandomWeighting (class in torchjd.aggregation) reset() (torchjd.aggregation.GradVac method) (torchjd.aggregation.GradVacWeighting method) (torchjd.aggregation.NashMTL method) (torchjd.aggregation.Stateful method)

S

Stateful (class in torchjd.aggregation)

Sum (class in torchjd.aggregation) SumWeighting (class in torchjd.aggregation)

T

torchjd.aggregation module torchjd.autogram module

torchjd.autojac module TrimmedMean (class in torchjd.aggregation)

U

UPGrad (class in torchjd.aggregation)

UPGradWeighting (class in torchjd.aggregation)

W

Weighting (class in torchjd.aggregation)