Commit Graph

278 Commits

Author SHA1 Message Date
CircleCI Docs
bd10c96fd8 rebase 2025-10-17 19:14:58 +00:00
CircleCI Docs
2d528f6927 rebase 2025-10-17 19:14:58 +00:00
Awni Hannun
6555eafe7b rebase 2025-10-17 19:14:58 +00:00
Awni Hannun
9acc86b81b docs update 2025-10-17 19:14:57 +00:00
Awni Hannun
50abfc9543 docs up 2025-10-17 19:14:57 +00:00
Awni Hannun
3b72c27899 docs update 2025-10-17 19:14:57 +00:00
Awni Hannun
44a49282c9 docs update 2025-10-17 19:14:57 +00:00
Awni Hannun
ee4f4e3816 docs update 2025-10-17 19:14:57 +00:00
Awni Hannun
9f4b63abe7 docs update 2025-10-17 19:14:57 +00:00
Awni Hannun
bea7b91017 docs update 2025-10-17 19:14:56 +00:00
Awni Hannun
098d163a6e docs update 2025-10-17 19:14:56 +00:00
Awni Hannun
4f6d88f179 docs 2025-10-17 19:14:56 +00:00
Awni Hannun
a7d2c589b2 docs update 2025-10-17 19:14:56 +00:00
Awni Hannun
db790ac051 docs update 2025-10-17 19:14:56 +00:00
Awni Hannun
3f707545f9 docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
b6f87dfaaf docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
4fbb58da79 docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
2c31a06405 use proper version 2025-10-17 19:14:55 +00:00
Awni Hannun
ba2f0b1e8e docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
2bc2b00a93 docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
d8f3a8a834 docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
3ee92b5d61 docs update 2025-10-17 19:14:55 +00:00
Awni Hannun
b8691a1637 remove uneeded files in docs 2025-10-17 19:14:55 +00:00
Awni Hannun
cfb54b960a update docs 2025-10-17 19:14:55 +00:00
Awni Hannun
03a66f24b0 docs update 2025-10-17 19:14:54 +00:00
Awni Hannun
cc06c8bc0e docs up 2025-10-17 19:14:54 +00:00
Awni Hannun
f5cfadc3d7 docs up 2025-10-17 19:14:54 +00:00
Awni Hannun
bd5f469bac docs update 2025-10-17 19:14:54 +00:00
Awni Hannun
6f71a74c87 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
20a3e22ff0 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
0f04ebb557 update docs 2025-10-17 19:14:54 +00:00
Awni Hannun
f6ae46f713 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
eb64c60144 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
86bd60c849 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
7a235ce49e docs 2025-10-17 19:14:54 +00:00
Awni Hannun
13e8d87a88 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
e09a97e24e docs 2025-10-17 19:14:54 +00:00
Awni Hannun
aa0e5e9049 docs 2025-10-17 19:14:54 +00:00
Awni Hannun
1c70552af7 docs 2025-10-17 19:14:53 +00:00
Awni Hannun
a372a3844d docs 2025-10-17 19:14:53 +00:00
Umberto Mignozzetti
3f730e77aa Update export function example for array input (#2598)
After changing the shape to conform (same shapes for all objects), the example works.
2025-09-16 14:38:05 -07:00
Umberto Mignozzetti
8afb6d62f2 Fix typo in average_gradients function call (#2594) 2025-09-15 11:29:21 -07:00
Umberto Mignozzetti
36cad99a11 Refactor code examples to use 'gelu' (#2592)
Updated code examples to use 'gelu' directly instead of 'nn.gelu'.
2025-09-15 09:47:02 -07:00
Gökdeniz Gülmez
db5443e831 Adding Relu2 (#2582)
* in. com.

* upd. ackn.

* update __init__

* nits

* nits + format

* used mx.maximum(x, 0) instead of calling the function and moves relu6 under relu2 to make it nicer

* same with _make_activation_module

* Update python/mlx/nn/layers/activations.py

upd

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

* update funct.rst

* upd. layers.rst

---------

Co-authored-by: Awni Hannun <awni.hannun@gmail.com>
2025-09-10 07:24:30 -07:00
Krishi Saripalli
c5d2937aa5 chore: Update Docs With Slice Copy Example (#2559)
* chore: updated docs with slice copy example

* nits

---------

Co-authored-by: Awni Hannun <awni@apple.com>
2025-09-02 22:07:02 -07:00
Awni Hannun
70560b6bd5 Add mode parameter for quantization (#2499)
* add mode parameter for quantization

* mxfp4 quantize/dequantize + start of optional biases

* mxfp4 works

* speedup

* cpu mxfp4

* fix

* fix test tol

* fix

* refactor

* add quant mode enum
2025-08-28 06:45:26 -07:00
Anastasiia Filippova
9392fc3f88 NCCL backend (#2476) 2025-08-21 11:56:15 -07:00
Angelos Katharopoulos
0c5fc63a36 Fix docs omission (#2524) 2025-08-20 17:56:06 -07:00
Angelos Katharopoulos
e397177f6e Custom cuda kernel (#2517) 2025-08-20 17:20:22 -07:00
russellizadi
512281781c Remove state return from function example in compile documentation (#2518) 2025-08-20 00:45:05 -07:00