zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-04 10:38:10 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity
1,476 Commits 42 Branches 73 Tags
ee4f4e3816064d92110c804cbb20c09ce0e3b658
Commit Graph

271 Commits

Author SHA1 Message Date
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
Awni Hannun
c5fcd5b61b fix custom kernel test (#2510) 2025-08-18 06:45:59 -07:00
Daniel Yeh
ac207ce7aa make code blocks copyable (#2480) 
Co-authored-by: Chen-Chen Yeh <ge96noj@mytum.de>
 2025-08-12 12:29:02 -07:00
Luca Vivona
728d4db582 Support destination arg in tree flatten/unflatten (#2450) 2025-08-06 15:34:59 -07:00
Awni Hannun
db5c7efcf6 revert default cuda install (#2465) 
* revert default cuda install

* revert default cuda install
 2025-08-06 06:19:12 -07:00
Awni Hannun
6ad0889c8a default install cuda on linux (#2462) 2025-08-04 15:33:05 -07:00
Awni Hannun
da5912e4f2 fix custom metal extension (#2446) 2025-07-31 06:25:36 -07:00
Awni Hannun
dcb8319f3d update install docs and requirements (#2419) 2025-07-25 12:13:19 -07:00