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
Gökdeniz Gülmez
deee214a95
Adding support for the Muon Optimizer ( #1914 )
...
* initial commit with workong optmimizer
* update ACKNOWLEDGMENTS.md
* nits and adding it to test
* nits
* G.astype(mx.bfloat16) to G.astype(G.dtype)
* G.ndim >= 2 to assert G.ndim == 2
* remove coments
* replace with mx.addmm
* remove comments
* format
* nits
* match muon
* fix addmm
---------
Co-authored-by: Awni Hannun <awni@apple.com >
2025-07-18 12:25:28 -07:00
Awni Hannun
f0a0b077a0
Install linux with mlx[cuda] and mlx[cpu] ( #2356 )
...
* install linux with mlx[cuda] and mlx[cpu]
* temp for testing
* cleanup circle, fix cuda repair
* update circle
* update circle
* decouple python bindings from core libraries
2025-07-14 17:17:33 -07:00
Cheng
d34f887abc
Add Primitive::name and remove Primitive::print ( #2365 )
2025-07-14 14:06:35 -07:00
Cheng
19facd4b20
Build with all cpu cores by default ( #2336 )
2025-07-07 06:06:45 -07:00
Awni Hannun
76831ed83d
Build CUDA release in Circle ( #2306 )
...
* cuda release
* add license
2025-06-19 15:26:36 -07:00
Awni Hannun
c552ff2451
[CUDA] Fix back-end bugs and enable corresponding tests ( #2296 )
...
* Fix some cuda back-end bugs and enable corresponding tests
* more fixes
* enable more tests
* format
2025-06-16 08:45:40 -07:00
Awni Hannun
1ca616844b
Fix unintuitive metal kernel caching ( #2242 )
...
* Fix unintuitive metal kernel caching
* alternative solution
2025-06-06 20:08:15 -07:00
Awni Hannun
55b4062dd8
copyright in docs ( #2214 )
2025-05-21 17:13:04 -07:00
Awni Hannun
a2cadb8218
real and imag properties ( #2189 )
2025-05-15 18:17:50 -07:00
Awni Hannun
c1eb9d05d9
non-symmetric eig and eigh ( #2188 )
2025-05-15 13:01:44 -07:00
Aashiq Dheeraj
bb6565ef14
add fftshift and ifftshift fft helpers ( #2135 )
...
* add fftshift and ifftshift fft helpers
* address comments
* axes have to be iterable
* fix fp error in roll + add test
---------
Co-authored-by: Aashiq Dheeraj <aashiq@aashiq-mbp-m4.local >
2025-04-29 22:13:45 -07:00
Yury Popov
e9e268336b
LogCumSumExp ( #2069 )
2025-04-13 01:27:29 -07:00
