Code

Data parallel helper#1407

Code

Merged

angeloskath merged 3 commits intomainml-explore/mlx:mainfrom data-parallel-helperml-explore/mlx:data-parallel-helperCopy head branch name to clipboardSep 17, 2024

+213-7

Commits

Commits on Sep 12, 2024

• Add float16 reduction to MPI

  angeloskath

  committed
  e0d6f2e

• Provide a helper to average gradients

  angeloskath

  committed
  27a0b02

Commits on Sep 16, 2024

• Remove sort from tree_flatten

  angeloskath

  committed
  861c689