Angelos Katharopoulos e397177f6e Custom cuda kernel (#2517)

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

..

nn

Doc fix (#1615)

2024-11-22 11:12:25 -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

Fp64 on the CPU (#1843)

2025-02-07 15:52:22 -08:00

devices_and_streams.rst

Add synchronize function (#1006)

2024-04-22 08:25:46 -07: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

Ring docs (#1829)

2025-02-28 11:34:21 -08:00

ops.rst

LogCumSumExp (#2069)

2025-04-13 01:27:29 -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

Add missing funcs to docs (#2021)

2025-03-30 18:29:33 -07:00

tree_utils.rst

feat: implement clip_grad_norm (#1043)

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