Repository navigation

[TPU] Add tensor parallel support for TPU backend - #14075

Closed

JingyaHuang wants to merge 45 commits into

huggingface:mainfrom

JingyaHuang:extend-tp-to-tpu

Commits on Mar 18, 2026
- draft:add neuron as a legit backend
  JingyaHuang
  committed
  98f6c8c
- Merge branch 'huggingface:main' into add-neuron-backend
  JingyaHuang
  authored
  c58b8b8
Commits on Mar 19, 2026
- Merge branch 'huggingface:main' into add-neuron-backend
  JingyaHuang
  authored
  3367409
Commits on Mar 25, 2026
- Merge branch 'main' into add-neuron-backend
  JingyaHuang
  authored
  0c51734
Commits on Mar 26, 2026
- feat: neuron-specific changes in the pipeline
  JingyaHuang
  committed
  a76953c
Commits on Mar 27, 2026
- tests: eager tests
  JingyaHuang
  committed
  2480388
Commits on Apr 9, 2026
- draft: start with tp for flux2
  JingyaHuang
  committed
  1469c04
- fix: style
  JingyaHuang
  committed
  929ab72
- Merge branch 'huggingface:main' into add-neuron-backend
  JingyaHuang
  authored
  52cac76
- Merge branch 'huggingface:main' into support-neuron-tp
  JingyaHuang
  authored
  30cb353
- Merge branch 'add-neuron-backend' of github.com:JingyaHuang/diffusers into add-neuron-backend
  JingyaHuang
  committed
  28a5086
Commits on Apr 10, 2026
- Merge branch 'huggingface:main' into support-neuron-tp
  JingyaHuang
  authored
  7fab0c4
- Merge branch 'huggingface:main' into add-neuron-backend
  JingyaHuang
  authored
  68689e5
- Merge branch 'main' into add-neuron-backend
  JingyaHuang
  authored
  da79308
- fix:apr_02 beta
  JingyaHuang
  committed
  3bb9c7c
- Merge branch 'add-neuron-backend' of github.com:JingyaHuang/diffusers into add-neuron-backend
  JingyaHuang
  committed
  c4facab
- feat:add wan
  JingyaHuang
  committed
  dff1f32
Commits on Apr 13, 2026
- Merge branch 'huggingface:main' into support-neuron-tp
  JingyaHuang
  authored
  1c930c4
- Merge branch 'huggingface:main' into add-neuron-backend
  JingyaHuang
  authored
  1eb5ff9
Commits on Apr 14, 2026
- fix:pixart
  JingyaHuang
  committed
  cbe8f28
Commits on Apr 15, 2026
- fix: rewrite flux swiglu activation to avoid gather op in neuron IR
  JingyaHuang
  committed
  16b9606
- test: pixart compile mode on neuron
  JingyaHuang
  committed
  7f13f68
Commits on Apr 22, 2026
- Merge branch 'main' into neuron-torch-comppile
  JingyaHuang
  authored
  a46cb19
Commits on May 11, 2026
- cleanup & fix style
  JingyaHuang
  committed
  a354b88
- Merge branch 'neuron-torch-comppile' into support-neuron-tp
  JingyaHuang
  committed
  931bb85
- Merge branch 'main' into support-neuron-tp
  JingyaHuang
  authored
  9ab6dc3
Commits on Jun 22, 2026
- Merge branch 'main' into support-neuron-tp
  JingyaHuang
  committed
  48fb75b
- merge: another change
  JingyaHuang
  committed
  c350f7b
- Merge branch 'main' into support-neuron-tp
  JingyaHuang
  authored
  644477a
Commits on Jun 23, 2026
- review: cleanup+suggestions
  JingyaHuang
  committed
  03cb725
- Merge branch 'support-neuron-tp' of github.com:JingyaHuang/diffusers into support-neuron-tp
  JingyaHuang
  committed
  9da93ed
Commits on Jun 24, 2026
- fix: CIs style
  JingyaHuang
  committed
  d44f772
- Merge branch 'main' into support-neuron-tp
  JingyaHuang
  authored
  3fc043e
- tests: add test units for tp
  JingyaHuang
  committed
  e6d20d8
- Merge branch 'support-neuron-tp' of github.com:JingyaHuang/diffusers into support-neuron-tp
  JingyaHuang
  committed
  e76a2fc
- fix: in case of text-encoder(s) on CPU
  JingyaHuang
  committed
  034ba9e
Commits on Jun 25, 2026
- review:cleanup+add test
  JingyaHuang
  committed
  4907524
- Merge branch 'support-neuron-tp' of github.com:JingyaHuang/diffusers into support-neuron-tp
  JingyaHuang
  committed
  af2aed7
- Merge branch 'main' into support-neuron-tp
  JingyaHuang
  authored
  b9b048b
- fix: style
  JingyaHuang
  committed
  915eeb1
- Merge branch 'support-neuron-tp' of github.com:JingyaHuang/diffusers into support-neuron-tp
  JingyaHuang
  committed
  720dad2
Commits on Jun 26, 2026
- feat: add TP for TPU
  JingyaHuang
  committed
  7dfdb78
Commits on Jul 17, 2026
- Merge branch 'main' into extend-tp-to-tpu
  JingyaHuang
  authored
  ffaf477
- style: fix import sorting in TPU test scripts
  
  JingyaHuang
  and
  claude
  committed
  f1fe3e8
- style: ruff format TPU test scripts
  
  JingyaHuang
  and
  claude
  committed
  556b54a

Commits

Commits on Mar 18, 2026

Commits on Mar 19, 2026

Commits on Mar 25, 2026

Commits on Mar 26, 2026

Commits on Mar 27, 2026

Commits on Apr 9, 2026

Commits on Apr 10, 2026

Commits on Apr 13, 2026

Commits on Apr 14, 2026

Commits on Apr 15, 2026

Commits on Apr 22, 2026

Commits on May 11, 2026

Commits on Jun 22, 2026

Commits on Jun 23, 2026

Commits on Jun 24, 2026

Commits on Jun 25, 2026

Commits on Jun 26, 2026

Commits on Jul 17, 2026