Merge branch 'main' into refactor-stateful

Author: ValerianRey

CHANGELOG.md

@@ -8,10 +8,13 @@ changelog does not include internal changes that do not affect the user.
 
 ## [Unreleased]
 
+## [0.10.0] - 2026-04-16
+
 ### Added
 
 - Added `GradVac` and `GradVacWeighting` from
   [Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Models](https://arxiv.org/pdf/2010.05874).
+- Documented per-parameter-group aggregation (GradVac-style grouping) in a new Grouping example.
 
 ### Fixed
 

README.md

@@ -281,7 +281,7 @@ TorchJD provides many existing aggregators from the literature, listed in the fo
 | [Constant](https://torchjd.org/stable/docs/aggregation/constant#torchjd.aggregation.Constant)              | [ConstantWeighting](https://torchjd.org/stable/docs/aggregation/constant#torchjd.aggregation.ConstantWeighting)        | -                                                                                                                                                                    |
 | [DualProj](https://torchjd.org/stable/docs/aggregation/dualproj#torchjd.aggregation.DualProj)              | [DualProjWeighting](https://torchjd.org/stable/docs/aggregation/dualproj#torchjd.aggregation.DualProjWeighting)        | [Gradient Episodic Memory for Continual Learning](https://arxiv.org/pdf/1706.08840)                                                                                  |
 | [GradDrop](https://torchjd.org/stable/docs/aggregation/graddrop#torchjd.aggregation.GradDrop)              | -                                                                                                                      | [Just Pick a Sign: Optimizing Deep Multitask Models with Gradient Sign Dropout](https://arxiv.org/pdf/2010.06808)                                                    |
-| [GradVac](https://torchjd.org/latest/docs/aggregation/gradvac#torchjd.aggregation.GradVac)              | [GradVacWeighting](https://torchjd.org/latest/docs/aggregation/gradvac#torchjd.aggregation.GradVacWeighting)                                                                                                                      | [Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Models](https://arxiv.org/pdf/2010.05874)                                                    |
+| [GradVac](https://torchjd.org/stable/docs/aggregation/gradvac#torchjd.aggregation.GradVac)              | [GradVacWeighting](https://torchjd.org/stable/docs/aggregation/gradvac#torchjd.aggregation.GradVacWeighting)                                                                                                                      | [Gradient Vaccine: Investigating and Improving Multi-task Optimization in Massively Multilingual Models](https://arxiv.org/pdf/2010.05874)                                                    |
 | [IMTLG](https://torchjd.org/stable/docs/aggregation/imtl_g#torchjd.aggregation.IMTLG)                      | [IMTLGWeighting](https://torchjd.org/stable/docs/aggregation/imtl_g#torchjd.aggregation.IMTLGWeighting)                | [Towards Impartial Multi-task Learning](https://discovery.ucl.ac.uk/id/eprint/10120667/)                                                                             |
 | [Krum](https://torchjd.org/stable/docs/aggregation/krum#torchjd.aggregation.Krum)                          | [KrumWeighting](https://torchjd.org/stable/docs/aggregation/krum#torchjd.aggregation.KrumWeighting)                    | [Machine Learning with Adversaries: Byzantine Tolerant Gradient Descent](https://proceedings.neurips.cc/paper/2017/file/f4b9ec30ad9f68f89b29639786cb62ef-Paper.pdf)  |
 | [Mean](https://torchjd.org/stable/docs/aggregation/mean#torchjd.aggregation.Mean)                          | [MeanWeighting](https://torchjd.org/stable/docs/aggregation/mean#torchjd.aggregation.MeanWeighting)                    | -                                                                                                                                                                    |

docs/source/conf.py

@@ -94,7 +94,11 @@ def linkcode_resolve(domain: str, info: dict[str, str]) -> str | None:
     if domain != "py" or not info["module"]:
         return None
 
-    obj = _get_obj(info)
+    try:
+        obj = _get_obj(info)
+    except AttributeError:
+        # This can happen when trying to get a field typed at the class level.
+        return None
     file_name = _get_file_name(obj)
 
     if not file_name:

docs/source/docs/aggregation/aligned_mtl.rst

@@ -4,11 +4,5 @@ Aligned-MTL
 ===========
 
 .. autoclass:: torchjd.aggregation.AlignedMTL
-    :members:
-    :undoc-members:
-    :exclude-members: forward
 
 .. autoclass:: torchjd.aggregation.AlignedMTLWeighting
-    :members:
-    :undoc-members:
-    :exclude-members: forward

docs/source/docs/aggregation/cagrad.rst

@@ -4,11 +4,5 @@ CAGrad
 ======
 
 .. autoclass:: torchjd.aggregation.CAGrad
-    :members:
-    :undoc-members:
-    :exclude-members: forward
 
 .. autoclass:: torchjd.aggregation.CAGradWeighting
-    :members:
-    :undoc-members:
-    :exclude-members: forward

docs/source/docs/aggregation/config.rst

@@ -4,6 +4,3 @@ ConFIG
 ======
 
 .. autoclass:: torchjd.aggregation.ConFIG
-    :members:
-    :undoc-members:
-    :exclude-members: forward

docs/source/docs/aggregation/constant.rst

@@ -4,11 +4,5 @@ Constant
 ========
 
 .. autoclass:: torchjd.aggregation.Constant
-    :members:
-    :undoc-members:
-    :exclude-members: forward
 
 .. autoclass:: torchjd.aggregation.ConstantWeighting
-    :members:
-    :undoc-members:
-    :exclude-members: forward

docs/source/docs/aggregation/dualproj.rst

@@ -4,11 +4,5 @@ DualProj
 ========
 
 .. autoclass:: torchjd.aggregation.DualProj
-    :members:
-    :undoc-members:
-    :exclude-members: forward
 
 .. autoclass:: torchjd.aggregation.DualProjWeighting
-    :members:
-    :undoc-members:
-    :exclude-members: forward

docs/source/docs/aggregation/flattening.rst

@@ -4,6 +4,3 @@ Flattening
 ==========
 
 .. autoclass:: torchjd.aggregation.Flattening
-    :members:
-    :undoc-members:
-    :exclude-members: forward

docs/source/docs/aggregation/graddrop.rst

@@ -4,6 +4,3 @@ GradDrop
 ========
 
 .. autoclass:: torchjd.aggregation.GradDrop
-    :members:
-    :undoc-members:
-    :exclude-members: forward