Skip to content

Commit f51a2f2

Browse files
committed
1 parent 8410ece commit f51a2f2

44 files changed

Lines changed: 160 additions & 162 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

stable/_sources/docs/autogram/index.rst.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
autogram
22
========
33

4-
54
.. automodule:: torchjd.autogram
6-
:members:
5+
:no-members:
76

87
.. toctree::
98
:hidden:

stable/_sources/docs/autojac/index.rst.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@ autojac
22
=======
33

44
.. automodule:: torchjd.autojac
5-
:members:
6-
5+
:no-members:
76

87
.. toctree::
98
:hidden:

stable/_sources/examples/monitoring.rst.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ they have a negative inner product).
4949
optimizer = SGD(params, lr=0.1)
5050
aggregator = UPGrad()
5151
52-
aggregator.weighting.register_forward_hook(print_weights)
52+
aggregator.weighting.weighting.register_forward_hook(print_weights)
5353
aggregator.register_forward_hook(print_gd_similarity)
5454
5555
inputs = torch.randn(8, 16, 10) # 8 batches of 16 random input vectors of length 10

stable/_static/documentation_options.js

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
const DOCUMENTATION_OPTIONS = {
2-
VERSION: '0.8.0',
2+
VERSION: '0.8.1',
33
LANGUAGE: 'en',
44
COLLAPSE_INDEX: false,
55
BUILDER: 'dirhtml',

stable/_static/styles/furo.css

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

stable/_static/styles/furo.css.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

stable/docs/aggregation/aligned_mtl/index.html

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@
88
<link rel="prefetch" href="../../../_static/logo-light-mode.png" as="image">
99
<link rel="prefetch" href="../../../_static/logo-dark-mode.png" as="image">
1010

11-
<link rel="shortcut icon" href="../../../_static/favicon.ico"><!-- Generated with Sphinx 8.2.3 and Furo 2025.09.25 -->
11+
<link rel="shortcut icon" href="../../../_static/favicon.ico"><!-- Generated with Sphinx 8.2.3 and Furo 2025.12.19 -->
1212
<title>Aligned-MTL - TorchJD</title>
1313
<link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=d111a655" />
14-
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo.css?v=580074bf" />
14+
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo.css?v=7bdb33bb" />
1515
<link rel="stylesheet" type="text/css" href="../../../_static/sphinx-design.min.css?v=95c83b7e" />
1616
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo-extensions.css?v=8dab3a3b" />
1717

@@ -293,7 +293,7 @@
293293
<h1>Aligned-MTL<a class="headerlink" href="#aligned-mtl" title="Link to this heading"></a></h1>
294294
<dl class="py class">
295295
<dt class="sig sig-object py" id="torchjd.aggregation.AlignedMTL">
296-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">AlignedMTL</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pref_vector</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.0/src/torchjd/aggregation/_aligned_mtl.py#L37-L59"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.AlignedMTL" title="Link to this definition"></a></dt>
296+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">AlignedMTL</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pref_vector</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.1/src/torchjd/aggregation/_aligned_mtl.py#L37-L59"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.AlignedMTL" title="Link to this definition"></a></dt>
297297
<dd><p><a class="reference internal" href="../#torchjd.aggregation.Aggregator" title="torchjd.aggregation._aggregator_bases.Aggregator"><code class="xref py py-class docutils literal notranslate"><span class="pre">Aggregator</span></code></a> as defined in Algorithm 1 of
298298
<a class="reference external" href="https://openaccess.thecvf.com/content/CVPR2023/papers/Senushkin_Independent_Component_Alignment_for_Multi-Task_Learning_CVPR_2023_paper.pdf">Independent Component Alignment for Multi-Task Learning</a>.</p>
299299
<dl class="field-list simple">
@@ -310,7 +310,7 @@ <h1>Aligned-MTL<a class="headerlink" href="#aligned-mtl" title="Link to this hea
310310

311311
<dl class="py class">
312312
<dt class="sig sig-object py" id="torchjd.aggregation.AlignedMTLWeighting">
313-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">AlignedMTLWeighting</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pref_vector</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.0/src/torchjd/aggregation/_aligned_mtl.py#L62-L99"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.AlignedMTLWeighting" title="Link to this definition"></a></dt>
313+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">AlignedMTLWeighting</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pref_vector</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.1/src/torchjd/aggregation/_aligned_mtl.py#L62-L99"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.AlignedMTLWeighting" title="Link to this definition"></a></dt>
314314
<dd><p><a class="reference internal" href="../#torchjd.aggregation.Weighting" title="torchjd.aggregation._weighting_bases.Weighting"><code class="xref py py-class docutils literal notranslate"><span class="pre">Weighting</span></code></a> giving the weights of
315315
<a class="reference internal" href="#torchjd.aggregation.AlignedMTL" title="torchjd.aggregation.AlignedMTL"><code class="xref py py-class docutils literal notranslate"><span class="pre">AlignedMTL</span></code></a>.</p>
316316
<dl class="field-list simple">
@@ -372,7 +372,7 @@ <h1>Aligned-MTL<a class="headerlink" href="#aligned-mtl" title="Link to this hea
372372

373373
</aside>
374374
</div>
375-
</div><script src="../../../_static/documentation_options.js?v=0c033e68"></script>
375+
</div><script src="../../../_static/documentation_options.js?v=1ec50272"></script>
376376
<script src="../../../_static/doctools.js?v=9bcbadda"></script>
377377
<script src="../../../_static/sphinx_highlight.js?v=dc90522c"></script>
378378
<script src="../../../_static/scripts/furo.js?v=46bd48cc"></script>

stable/docs/aggregation/cagrad/index.html

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@
88
<link rel="prefetch" href="../../../_static/logo-light-mode.png" as="image">
99
<link rel="prefetch" href="../../../_static/logo-dark-mode.png" as="image">
1010

11-
<link rel="shortcut icon" href="../../../_static/favicon.ico"><!-- Generated with Sphinx 8.2.3 and Furo 2025.09.25 -->
11+
<link rel="shortcut icon" href="../../../_static/favicon.ico"><!-- Generated with Sphinx 8.2.3 and Furo 2025.12.19 -->
1212
<title>CAGrad - TorchJD</title>
1313
<link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=d111a655" />
14-
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo.css?v=580074bf" />
14+
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo.css?v=7bdb33bb" />
1515
<link rel="stylesheet" type="text/css" href="../../../_static/sphinx-design.min.css?v=95c83b7e" />
1616
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo-extensions.css?v=8dab3a3b" />
1717

@@ -293,7 +293,7 @@
293293
<h1>CAGrad<a class="headerlink" href="#cagrad" title="Link to this heading"></a></h1>
294294
<dl class="py class">
295295
<dt class="sig sig-object py" id="torchjd.aggregation.CAGrad">
296-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">CAGrad</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">c</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">norm_eps</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.0001</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.0/src/torchjd/aggregation/_cagrad.py#L18-L47"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.CAGrad" title="Link to this definition"></a></dt>
296+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">CAGrad</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">c</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">norm_eps</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.0001</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.1/src/torchjd/aggregation/_cagrad.py#L18-L47"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.CAGrad" title="Link to this definition"></a></dt>
297297
<dd><p><a class="reference internal" href="../#torchjd.aggregation.Aggregator" title="torchjd.aggregation._aggregator_bases.Aggregator"><code class="xref py py-class docutils literal notranslate"><span class="pre">Aggregator</span></code></a> as defined in Algorithm 1 of
298298
<a class="reference external" href="https://arxiv.org/pdf/2110.14048.pdf">Conflict-Averse Gradient Descent for Multi-task Learning</a>.</p>
299299
<dl class="field-list simple">
@@ -315,7 +315,7 @@ <h1>CAGrad<a class="headerlink" href="#cagrad" title="Link to this heading">¶</
315315

316316
<dl class="py class">
317317
<dt class="sig sig-object py" id="torchjd.aggregation.CAGradWeighting">
318-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">CAGradWeighting</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">c</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">norm_eps</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.0001</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.0/src/torchjd/aggregation/_cagrad.py#L50-L103"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.CAGradWeighting" title="Link to this definition"></a></dt>
318+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">CAGradWeighting</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">c</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">norm_eps</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">0.0001</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.1/src/torchjd/aggregation/_cagrad.py#L50-L103"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.CAGradWeighting" title="Link to this definition"></a></dt>
319319
<dd><p><a class="reference internal" href="../#torchjd.aggregation.Weighting" title="torchjd.aggregation._weighting_bases.Weighting"><code class="xref py py-class docutils literal notranslate"><span class="pre">Weighting</span></code></a> giving the weights of
320320
<a class="reference internal" href="#torchjd.aggregation.CAGrad" title="torchjd.aggregation.CAGrad"><code class="xref py py-class docutils literal notranslate"><span class="pre">CAGrad</span></code></a>.</p>
321321
<dl class="field-list simple">
@@ -386,7 +386,7 @@ <h1>CAGrad<a class="headerlink" href="#cagrad" title="Link to this heading">¶</
386386

387387
</aside>
388388
</div>
389-
</div><script src="../../../_static/documentation_options.js?v=0c033e68"></script>
389+
</div><script src="../../../_static/documentation_options.js?v=1ec50272"></script>
390390
<script src="../../../_static/doctools.js?v=9bcbadda"></script>
391391
<script src="../../../_static/sphinx_highlight.js?v=dc90522c"></script>
392392
<script src="../../../_static/scripts/furo.js?v=46bd48cc"></script>

stable/docs/aggregation/config/index.html

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,10 @@
88
<link rel="prefetch" href="../../../_static/logo-light-mode.png" as="image">
99
<link rel="prefetch" href="../../../_static/logo-dark-mode.png" as="image">
1010

11-
<link rel="shortcut icon" href="../../../_static/favicon.ico"><!-- Generated with Sphinx 8.2.3 and Furo 2025.09.25 -->
11+
<link rel="shortcut icon" href="../../../_static/favicon.ico"><!-- Generated with Sphinx 8.2.3 and Furo 2025.12.19 -->
1212
<title>ConFIG - TorchJD</title>
1313
<link rel="stylesheet" type="text/css" href="../../../_static/pygments.css?v=d111a655" />
14-
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo.css?v=580074bf" />
14+
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo.css?v=7bdb33bb" />
1515
<link rel="stylesheet" type="text/css" href="../../../_static/sphinx-design.min.css?v=95c83b7e" />
1616
<link rel="stylesheet" type="text/css" href="../../../_static/styles/furo-extensions.css?v=8dab3a3b" />
1717

@@ -293,7 +293,7 @@
293293
<h1>ConFIG<a class="headerlink" href="#config" title="Link to this heading"></a></h1>
294294
<dl class="py class">
295295
<dt class="sig sig-object py" id="torchjd.aggregation.ConFIG">
296-
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">ConFIG</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pref_vector</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.0/src/torchjd/aggregation/_config.py#L37-L74"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.ConFIG" title="Link to this definition"></a></dt>
296+
<em class="property"><span class="k"><span class="pre">class</span></span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">torchjd.aggregation.</span></span><span class="sig-name descname"><span class="pre">ConFIG</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pref_vector</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="reference external" href="https://github.com/TorchJD/torchjd/blob/v0.8.1/src/torchjd/aggregation/_config.py#L37-L74"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#torchjd.aggregation.ConFIG" title="Link to this definition"></a></dt>
297297
<dd><p><a class="reference internal" href="../#torchjd.aggregation.Aggregator" title="torchjd.aggregation._aggregator_bases.Aggregator"><code class="xref py py-class docutils literal notranslate"><span class="pre">Aggregator</span></code></a> as defined in Equation 2 of <a class="reference external" href="https://arxiv.org/pdf/2408.11104">ConFIG:
298298
Towards Conflict-free Training of Physics Informed Neural Networks</a>.</p>
299299
<dl class="field-list simple">
@@ -359,7 +359,7 @@ <h1>ConFIG<a class="headerlink" href="#config" title="Link to this heading">¶</
359359

360360
</aside>
361361
</div>
362-
</div><script src="../../../_static/documentation_options.js?v=0c033e68"></script>
362+
</div><script src="../../../_static/documentation_options.js?v=1ec50272"></script>
363363
<script src="../../../_static/doctools.js?v=9bcbadda"></script>
364364
<script src="../../../_static/sphinx_highlight.js?v=dc90522c"></script>
365365
<script src="../../../_static/scripts/furo.js?v=46bd48cc"></script>

0 commit comments

Comments
 (0)