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

Co-authored-by: Awni Hannun <awni@apple.com>

2026-01-28 17:14:56 -08:00

..

nn

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

2026-01-28 17:14:56 -08:00

optimizers

Adding support for the Muon Optimizer (#1914)

2025-07-18 12:25:28 -07:00

array.rst

real and imag properties (#2189)

2025-05-15 18:17:50 -07:00

cuda.rst

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

data_types.rst

Fix float64 size in data_types.rst (#2948)

2025-12-22 16:24:07 -08:00

devices_and_streams.rst

GPU discovery (#3055)

2026-01-26 09:54:13 -08:00

distributed.rst

Adds send/recv ops in distributed (#1366)

2024-08-26 23:01:37 -07:00

export.rst

Export / import functions to / from a file (#1642)

2024-12-24 11:19:13 -08:00

fast.rst

Custom cuda kernel (#2517)

2025-08-20 17:20:22 -07:00

fft.rst

add fftshift and ifftshift fft helpers (#2135)

2025-04-29 22:13:45 -07:00

linalg.rst

non-symmetric eig and eigh (#2188)

2025-05-15 13:01:44 -07:00

memory_management.rst

fix docs (#1991)

2025-03-21 19:58:53 -07:00

metal.rst

fix docs (#1991)

2025-03-21 19:58:53 -07:00

nn.rst

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

2026-01-28 17:14:56 -08:00

ops.rst

add median op (#2705)

2025-10-27 11:33:42 -07:00

optimizers.rst

Support destination arg in tree flatten/unflatten (#2450)

2025-08-06 15:34:59 -07:00

random.rst

Add mx.random.permutation (#1471)

2024-10-08 19:42:19 -07:00

transforms.rst

Type Enhancement for Func Transforms and Bug Fix (#3003)

2026-01-20 07:19:57 +09:00

tree_utils.rst

feat: implement clip_grad_norm (#1043)

2024-05-03 09:07:02 -07:00