Code

[WIP] Federated learning#2418

Code

Draft

maximegmd wants to merge 9 commits intopytorch:mainpytorch/torchtune:mainfrom maximegmd:add-feature-federated-learningmaximegmd/torchtune:add-feature-federated-learningCopy head branch name to clipboard

Commits

Commits on Feb 20, 2025

• Add base federation recipe

  maximegmd

  committed
  370e319

  Copy full SHA for 370e319

Commits on Feb 21, 2025

• Basic DiLiCo, tested with Qwen 2.5 - 0.5b

  maximegmd

  committed
  a59b14a

  Copy full SHA for a59b14a

• Typo

  maximegmd

  committed
  7a8b42f

  Copy full SHA for 7a8b42f

• Minor cleanup

  maximegmd

  committed
  f97757f

  Copy full SHA for f97757f

Commits on Feb 26, 2025

• Merge remote-tracking branch 'upstream/main' into add-feature-federated-learning

  maximegmd

  committed
  9a0d024

  Copy full SHA for 9a0d024

• Merge remote-tracking branch 'upstream/main' into add-feature-federated-learning

  maximegmd

  committed
  532071f

  Copy full SHA for 532071f

Commits on Feb 27, 2025

• preparing for distributed FL

  maximegmd

  committed
  d5dc4e6

  Copy full SHA for d5dc4e6

Commits on Mar 10, 2025

• Improve memory usage

  maximegmd

  committed
  aead7a4

  Copy full SHA for aead7a4

• Merge branch 'main' of https://github.com/pytorch/torchtune into add-feature-federated-learning

  maximegmd

  committed
  f16017e

  Copy full SHA for f16017e