docs(aggregation): Add hosted interactive plotter webapp by KhusPatel4450 · Pull Request #772 · SimplexLab/TorchJD · GitHub

Adds a self-contained Dash webapp in visualization/ that hosts the interactive gradient plotter, and a GitHub Actions workflow that auto-deploys it to Hugging Face Spaces on every push to main.

visualization/app.py, Dash app refactored from tests/plots/interactive_plotter.py
visualization/_utils.py, plotting utilities (copy of tests/plots/_utils.py)
visualization/requirements.txt + visualization/Dockerfile, for HF Spaces (Docker SDK)
visualization/README.md, HF Spaces metadata
.github/workflows/deploy-visualization.yml, deploys visualization/ to HF Spaces on push to main

The app accepts query parameters so individual aggregator pages can embed it with a pre-selected method (as suggested by @PierreQuinton)

Before we can deploy we need to do this:

Create a Hugging Face Space at torchjd/interactive-plotter with Docker as the SDK
Add HF_TOKEN (an HF token with write access to that Space) as a GitHub Actions secret

Once those are in place, any push to main that touches visualization/ will redeploy automatically

Thank you very much for taking this! Could you deploy it on under some random name on HF so that we can test it already?

Having docker sdk requires a paid hugging face plan. Is there a way to do it with the Static SDK? Or using another provider?

yeah you are right, it does required a paid plan, didn't realize that, anyway i am in the process of converting it to gradio, that one is free

okay looks like Gradio is paid too, I am just going to get the HF pro subscription, its pretty cheap, plus we can't do static because it doesn't support python

…bility Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

here is the space: https://huggingface.co/spaces/Hsuhk20/torchjd-plotter

I just tried: very cool!

What's the price of the HF pro subscription? Is there any risk of extra cost if people use this heavily (I've heard of people getting unexpected 300$ extra cost from HF by using their pro plan).

Should we create a HF pro account for SimplexLab and use it for that? Especially if the HF_TOKEN, with access to this account, has to be stored in the repo. Then it's not a good thing if it's linked to a personal account. And I'd rather not have you pay for this @KhusPatel4450 .

Suggested change

      
            the gradients. The green region shows the dual cone — any descent direction must lie inside it.
          
            the gradients. The green region shows the dual cone: the set of vectors with a
          
            non-negative inner product with each gradient.

Could we avoid code duplication between this and tests/plots/? I think we may want to move app.py, the newly added README.md and requirements.txt to tests/plots/. I still want to be able to run the thing locally (what interactive_plotter.py does) because it's more responsive to it's a good tool for us when working on new aggregators.

yeah I mean good point, this could be done easily, the GitHub Actions workflow would just need to point at tests/plots/. I am happy to restructure it if you think that's cleaner

Yes please go ahead!

Should we deploy only on new releases (it's when we may actually change the behavior of aggregators / release new aggregators to our users) + manually if needed (e.g. if we update the plotter) (manually = workflow_dispatch in github terminology)?

okay yeah I could apply this, I think that's a better trigger

This lines implies that we store a HF_TOKEN as a repo secret or something, right?

Yes, HF_TOKEN would be stored as a GitHub Actions repository secret (Under Settings)

Hello,

I am pretty sure those stories that you have heard happened because of leaving GPU hardware running, CPU Basic (what we use) is a flat included cost with Pro, not usage-based, the cost is $9 per month.

We could make it for simplex lab org too, I think it makes sense too, I don't really mind paying for it though

Move app.py, README.md, and requirements.txt from visualization/ into tests/plots/ so they share _utils.py with the existing plotters. Update the deploy workflow to point at tests/plots/. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

So cool. Is this live? Can I try it? I think we can include an HTML in Github md, so it would be nice to test here in a comment if it works. I think this would be a great add to the README.md at some point. I can pay for the HF, I have created a SimplexLab account, we can transfer at some point.

feat: Add hosted interactive plotter webapp

f95e0a2

KhusPatel4450 added the cc: feat Conventional commit type for new features. label Aug 1, 2026

KhusPatel4450 requested review from a team, PierreQuinton and ValerianRey as code owners August 1, 2026 16:36

ValerianRey added cc: docs Conventional commit type for changes to the documentation. package: aggregation and removed cc: feat Conventional commit type for new features. labels Aug 1, 2026

github-actions Bot changed the title feat: Add hosted interactive plotter webapp docs(aggregation): Add hosted interactive plotter webapp Aug 1, 2026

Converting Code to Gradio for Plotter Webapp

07a8f9e

fix: update Gradio to 6.x and add audioop-lts for Python 3.13 compati…

118fddc

…bility Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>

ValerianRey reviewed Aug 2, 2026

View reviewed changes

KhusPatel4450 and others added 3 commits August 2, 2026 18:14

ci: trigger plotter deploy on release and workflow_dispatch

d19f44c

ci: exclude tests/plots/app.py from ty type checking

5d69829

+                        python -c "
+                        import os
+                        from huggingface_hub import HfApi
+                        api = HfApi(token=os.environ['HF_TOKEN'])

+              Interactive visualization of gradient aggregation methods from [TorchJD](https://torchjd.org).
+              Adjust the angle and length of each gradient vector and select aggregators to see how they combine
+              the gradients. The green region shows the dual cone — any descent direction must lie inside it.

	the gradients. The green region shows the dual cone — any descent direction must lie inside it.
	the gradients. The green region shows the dual cone: the set of vectors with a
	non-negative inner product with each gradient.

+              on:
+                push:
+                  branches: [main]

docs(aggregation): Add hosted interactive plotter webapp by KhusPatel4450 · Pull Request #772 · SimplexLab/TorchJD · GitHub

Uh oh!

docs(aggregation): Add hosted interactive plotter webapp - #772

docs(aggregation): Add hosted interactive plotter webapp#772KhusPatel4450 wants to merge 6 commits intoSimplexLab:mainSimplexLab/TorchJD:mainfrom KhusPatel4450:feat/interactive-plotter-webappKhusPatel4450/TorchJD_CR_MOGM:feat/interactive-plotter-webappCopy head branch name to clipboard

Conversation

KhusPatel4450 commented Aug 1, 2026

Uh oh!

ValerianRey commented Aug 1, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

PierreQuinton commented Aug 2, 2026

Uh oh!

KhusPatel4450 commented Aug 2, 2026

Uh oh!

KhusPatel4450 commented Aug 2, 2026

Uh oh!

KhusPatel4450 commented Aug 2, 2026

Uh oh!

ValerianRey left a comment

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

Choose a reason for hiding this comment

Uh oh!

KhusPatel4450 commented Aug 2, 2026

Uh oh!

PierreQuinton commented Aug 10, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

3 participants

docs(aggregation): Add hosted interactive plotter webapp#772
KhusPatel4450 wants to merge 6 commits into
SimplexLab:mainfrom
KhusPatel4450:feat/interactive-plotter-webapp

ValerianRey commented Aug 1, 2026 •
edited
Loading

PierreQuinton commented Aug 10, 2026 •
edited
Loading