0938db7e54
Add determinant and sign-log-determinant functions to mlx.core.linalg ( #3416 )
...
Co-authored-by: Lucas Fernandes Martins <Lucas-Fernandes-Martins@users.noreply.github.com >
2026-05-05 09:06:23 +09:00
dec6b4d10f
ThreadLocalStream in C++ ( #3405 )
2026-04-15 15:46:11 -07:00
befe42d303
Add printoptions ( #3333 )
2026-04-01 22:24:48 -07:00
2105df91da
Add fftfreq, rfftfreq and scalar axes for fftshift/ifftshift ( #3298 )
2026-03-31 18:29:16 -07:00
939e425c7a
Decouple CommandEncoder from Device ( #3316 )
2026-04-01 08:51:17 +09:00
9ab3913567
logo files ( #3308 )
2026-03-24 15:08:06 -07:00
ea91bd02cf
update requirements for Macbook Neo ( #3257 )
2026-03-16 04:33:09 -07:00
1d44d913e6
docs: fix PyTorch to MLX conversion example ( #3265 )
2026-03-16 04:20:12 -07:00
f145ece976
Fix/missing libs in docs ( #3190 )
2026-03-03 09:53:18 -08:00
72e04f7fb7
[CUDA] Fsdp (easy) ( #3130 )
2026-03-01 23:29:09 +01:00
f2f2d16451
Export: preserve Dtype state values in export callback arguments ( #3145 )
...
Co-authored-by: Awni Hannun <awni@apple.com >
2026-02-19 08:07:28 -08:00
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
a828e769be
GPU discovery ( #3055 )
...
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2026-01-26 09:54:13 -08:00
617fd9cbbd
Use C++20 ( #3050 )
2026-01-24 08:48:41 +09:00
687508dd98
win: symbol exports and minor fixes ( #3024 )
...
Co-authored-by: Cheng <zcbenz@gmail.com >
2026-01-23 10:16:22 +09:00
46d0fdc5ec
Type Enhancement for Func Transforms and Bug Fix ( #3003 )
...
Co-authored-by: Awni Hannun <awni@apple.com >
2026-01-20 07:19:57 +09:00
3fe7794f22
Reverts changing the MLX_IBV_DEVICES to MLX_JACCL_DEVICES ( #2999 )
2026-01-14 15:44:17 -08:00
a8197795f5
replace MLX_IBV_COORDINATOR with MLX_JACCL_COORDINATOR ( #2986 )
2026-01-13 11:26:25 -08:00
d9b950eb2f
refactor: use time.perf_counter for consistent and accurate benchmarking ( #2943 )
2025-12-28 06:16:13 -08:00
26dfe4f651
Fetch nanobind with cmake ( #2949 )
2025-12-24 10:23:45 +09:00
af2fca5b74
Fix float64 size in data_types.rst ( #2948 )
2025-12-22 16:24:07 -08:00
1fc313db9d
Metal logging ( #2904 )
...
Co-authored-by: Awni Hannun <awni@apple.com >
2025-12-17 20:48:07 -08:00
9194ec20a8
Thunderbolt RDMA communications backend ( #2808 )
2025-12-17 11:27:54 -08:00
6b330eb2d5
DOC : Add compile state example ( #2910 )
2025-12-16 06:32:58 -08:00
f7a400951a
Fix docs: replace mx.random.randn with mx.random.normal ( #2890 )
2025-12-09 11:46:30 -08:00
39289ef025
[CUDA] Release build for cuda 13 ( #2872 )
2025-12-04 21:42:26 -08:00
8879ee00eb
Support more Numpy interfaces for masked_scatter ( #2832 )
2025-12-01 17:51:02 -08:00
b3825ac149
Add Masked Scatter ( #2663 )
...
Co-authored-by: Awni Hannun <awni@apple.com >
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com >
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2025-11-19 14:53:32 -08:00
eba6a9d163
Compatibility with pip-installed openmpi ( #2741 )
2025-11-07 16:58:31 -08:00
26ceb507eb
only build for macos 14 and up ( #2731 )
...
* only build for macos 14 and up
* bump metal cpp
2025-11-04 09:44:15 -08:00
d1e06117e8
bump python ( #2694 )
2025-10-27 11:34:31 -07:00
539d8322d1
add median op ( #2705 )
2025-10-27 11:33:42 -07:00
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
8afb6d62f2
Fix typo in average_gradients function call ( #2594 )
2025-09-15 11:29:21 -07:00
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
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
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
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
9392fc3f88
NCCL backend ( #2476 )
2025-08-21 11:56:15 -07:00
0c5fc63a36
Fix docs omission ( #2524 )
2025-08-20 17:56:06 -07:00
e397177f6e
Custom cuda kernel ( #2517 )
2025-08-20 17:20:22 -07:00
512281781c
Remove state return from function example in compile documentation ( #2518 )
2025-08-20 00:45:05 -07:00
c5fcd5b61b
fix custom kernel test ( #2510 )
2025-08-18 06:45:59 -07:00
ac207ce7aa
make code blocks copyable ( #2480 )
...
Co-authored-by: Chen-Chen Yeh <ge96noj@mytum.de >
2025-08-12 12:29:02 -07:00
728d4db582
Support destination arg in tree flatten/unflatten ( #2450 )
2025-08-06 15:34:59 -07:00
db5c7efcf6
revert default cuda install ( #2465 )
...
* revert default cuda install
* revert default cuda install
2025-08-06 06:19:12 -07:00
6ad0889c8a
default install cuda on linux ( #2462 )
2025-08-04 15:33:05 -07:00
da5912e4f2
fix custom metal extension ( #2446 )
2025-07-31 06:25:36 -07:00
dcb8319f3d
update install docs and requirements ( #2419 )
2025-07-25 12:13:19 -07:00
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
Abhilash Shankarampeta