[Docs] Simple example of using MLX distributed (#2973)

Co-authored-by: Awni Hannun <[email protected]>
2026-01-28 17:14:56 -08:00
parent 12e386f308
commit c86a9bced1
13 changed files with 387 additions and 85 deletions
@@ -10,6 +10,7 @@ Layers
   :template: nn-module-template.rst

   ALiBi
+   AllToShardedLinear
   AvgPool1d
   AvgPool2d
   AvgPool3d
@@ -46,8 +47,10 @@ Layers
   Mish
   MultiHeadAttention
   PReLU
+   QuantizedAllToShardedLinear
   QuantizedEmbedding
   QuantizedLinear
+   QuantizedShardedToAllLinear
   RMSNorm
   ReLU
   ReLU2
@@ -56,6 +59,7 @@ Layers
   RoPE
   SELU
   Sequential
+   ShardedToAllLinear
   Sigmoid
   SiLU
   SinusoidalPositionalEncoding