[Docs] Simple example of using MLX distributed#2973

Merged

awni merged 8 commits intoml-explore:mainfrom

Jan 29, 2026

Commits on Jan 28, 2026

docs(setup): add instructions for docs dev environment

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

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

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

stefpi
authored and
awni
committed
docs: cleanup up tp inference

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

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

stefpi
authored and
awni
committed
Nits
awni
committed