Code

Distributed layers#1270

Code

Draft

angeloskath wants to merge 6 commits intomainml-explore/mlx:mainfrom distributed-layersml-explore/mlx:distributed-layersCopy head branch name to clipboard

+484-1

Commits

Commits on Jan 15, 2025

• Add MPI barrier

  

  angeloskath

  authored and

  awni

  committed
  1ca5dc1

  Copy full SHA for 1ca5dc1

• Add the distributed linear layers

  

  angeloskath

  authored and

  awni

  committed
  7752f93

  Copy full SHA for 7752f93

• Add quantized distributed layers

  

  angeloskath

  authored and

  awni

  committed
  0b1f657

  Copy full SHA for 0b1f657

• Add distributed layers to nn top-level

  

  angeloskath

  authored and

  awni

  committed
  ae93bdd

  Copy full SHA for ae93bdd

• Fixes in distributed layers

  

  angeloskath

  authored and

  awni

  committed
  82f2c42

  Copy full SHA for 82f2c42

• fix rebase

  awni

  committed
  31ba022

  Copy full SHA for 31ba022