[Docs] Simple example of using MLX distributed - #2973

Merged

awni merged 8 commits into

ml-explore:mainfrom

Jan 29, 2026

Commits on Jan 28, 2026

docs(setup): add instructions for docs dev environment

stefpi
authored and
awni
committed
a3e7673
docs: added distributed layers to nn indexing

stefpi
authored and
awni
committed
afaed07
docs: added tensor parallelism example and layer descriptions

stefpi
authored and
awni
committed
f270658
docs: tensor parallel inference on llama

stefpi
authored and
awni
committed
50c2d71
docs: cleanup up tp inference

stefpi
authored and
awni
committed
074cb2c
docs: add nn/distributed to index routines and helpers

stefpi
authored and
awni
committed
8d93b91
docs: extract data parallel training into seperate example doc

stefpi
authored and
awni
committed
33d4211
Nits
awni
committed
8e9e63a