d01b83dfe7
Use nb::ndarray for checking arrays ( #3283 )
2026-03-25 10:44:54 +09:00
70a0da6fca
Use thread local storage for frontend compile cache ( #3280 )
2026-03-20 07:44:45 +09:00
5fa1a8d59f
Support indexing with any type which implmented __index__ ( #3210 )
2026-03-19 15:19:08 -07:00
21c11fc9b0
Create default random key lazily ( #3278 )
2026-03-19 20:22:52 +09:00
7bc61cceed
Slice update with operation ( #3266 )
2026-03-18 06:18:02 -07:00
8cd377b7db
Add the bartlett function ( #3155 )
2026-03-03 11:40:54 -08:00
1ce7118303
Fix ref leak in mx.save/load with file like object ( #3187 )
2026-03-02 16:55:29 -08:00
a8ba5ac3e0
Implement mlx.core.blackman ( #3136 )
2026-02-25 13:42:40 -08: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
06305022ab
Tensor scale nvfp4 ( #3022 )
2026-02-18 11:19:26 +01:00
360639c2df
Add the hamming window function ( #3135 )
2026-02-17 00:56:05 -08:00
3bbe87e6dc
Add hanning window function ( #3124 )
2026-02-16 09:44:49 -08:00
fcbdd05022
More useful error for large indices ( #3079 )
2026-01-29 13:02:39 -08:00
0c6a895ed7
Use lower-right causal mask alignment consistently ( #2967 )
...
Co-authored-by: Awni Hannun <awni.hannun@gmail.com >
2026-01-28 17:15:14 -08:00
a828e769be
GPU discovery ( #3055 )
...
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com >
2026-01-26 09:54:13 -08:00
b6aa03e5b8
Update pre-commit hooks and versions for clang-format, black, and isort ( #3059 )
2026-01-26 06:57:04 -08:00
fd27829efa
Build and test python package on Windows CI ( #3049 )
2026-01-24 07:22:36 +09:00
1d56dfdf59
Use higher precision for linspace with double ( #3029 )
2026-01-21 06:20:50 -08: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
d96a2bdf57
Fix python package install path in stubgen ( #3009 )
2026-01-19 09:34:02 +09:00
099dcc0f4c
Expose to/from fp8 in Python and don't auto-convert fp8 when loading from safetensors ( #2985 )
2026-01-13 15:48:21 -08:00
7b1c46982a
fix doc ( #2988 )
2026-01-12 13:33:26 -08:00
edab937248
Add asarray to __array_namespace__ ( #2966 )
2026-01-12 06:16:27 -08:00
26dfe4f651
Fetch nanobind with cmake ( #2949 )
2025-12-24 10:23:45 +09:00
fc19a08caa
Set install rpath of python bindings with cmake ( #2934 )
2025-12-19 16:43:00 +09:00
9194ec20a8
Thunderbolt RDMA communications backend ( #2808 )
2025-12-17 11:27:54 -08:00
4cf5b29fc5
qqmm ( #2789 )
...
Co-authored-by: root <root@bolt-t9a77vmteu-94s9t6ymth.bolt-pods.turi-bolt.svc.cluster.local >
Co-authored-by: root <root@bolt-5azkyvd8ga-kgfzk84y6m.bolt-pods.turi-bolt.svc.cluster.local >
Co-authored-by: root <root@bolt-y4nktpaecv-ssnx24rdha.bolt-pods.turi-bolt.svc.cluster.local >
2025-12-16 09:28:28 -08:00
81dfe5f137
Fix grad in place updates ( #2899 )
2025-12-11 14:44:58 -08:00
e1fee0074b
Update nanobind pin to most recent version ( #2896 )
2025-12-11 06:07:36 -08:00
9abb0b8123
Added support for pytree types that inherit from tuple and typing.namedtuple ( #2845 )
2025-12-04 11:06:45 -08:00
50d3914c67
Update gumbel function signature parameters ( #2868 )
2025-12-03 15:37:35 -08:00
cacbdbf995
Fix init from double ( #2861 )
2025-12-03 06:08:11 -08:00
8879ee00eb
Support more Numpy interfaces for masked_scatter ( #2832 )
2025-12-01 17:51:02 -08:00
b054838780
Added clarification to apply_fn parameter of apply_to_modules ( #2831 )
...
Co-authored-by: Awni Hannun <awni@apple.com >
2025-11-26 15:40:56 -08:00
1d4eacb737
Fix mx.core.linspace type annotation ( #2820 )
...
Co-authored-by: Awni Hannun <awni.hannun@gmail.com >
2025-11-24 14:15:08 -08:00
5b0f047226
Fix mx.core.load type annotation ( #2819 )
2025-11-22 11:09:44 -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
32b18d8b66
Use std::optional for mask_arr arg ( #2763 )
2025-11-17 10:43:33 +09:00
9320eb89a8
Fix dequantize python sig (dtype default) ( #2752 )
2025-11-11 09:55:24 -08:00
27778156dc
Nccl reduce scatter, all gather ( #2727 )
...
* Added reduce scatter and all gather for nccl
* fix unused import, delete unused file
* small fix
* deleted useless condition
* fixed comments
* fix bug in eval_gpu, renamed to sum_scatter, fix docs
* final fix docs
* remove and
* Update mlx/distributed/mpi/mpi.cpp
Co-authored-by: Awni Hannun <awni.hannun@gmail.com >
* fix broken set input output
* fixes set output
* typo
* fix typo
* no cpu, no gpu for reduce scatter
---------
Co-authored-by: Awni Hannun <awni.hannun@gmail.com >
2025-11-05 08:21:11 -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
ec72b44417
Add quantize/dequantize for mxfp8 and nvfp4 ( #2688 )
...
* Add quantize/dequantize slow path for mxfp8 and nvfp4
* fast cuda kernel for mx/nv quantization
* fallback for cuda < 12.8 (#2697 )
* format (#2700 )
* fix (#2701 )
* metal kernels
* docs
* fix jit
* add default bits and group sizes
* improve quant docs
* fix output type of mxfp4 matmuls
2025-10-28 16:23:12 -07:00
539d8322d1
add median op ( #2705 )
2025-10-27 11:33:42 -07:00
233384161e
Improved mx.split() docs ( #2689 )
...
* Improved mx.split() documentation
* Fix typo in docstring for array split function
* add example
---------
Co-authored-by: Awni Hannun <awni@apple.com >
2025-10-24 09:48:41 -07:00
e89e8b4272
Export with callback ( #2612 )
...
* export with callback
* export with callback
* Add types, fix kwarg ordering bug + test
* cleanup, test, fix
* typos
2025-10-08 19:24:33 -07:00
85a8824a8c
Fix cumulative operations when axis=None ( #2653 )
2025-10-08 15:25:38 -07:00
eb24267b56
Compile now can attach arbitrary data to an entry ( #2634 )
2025-09-30 13:33:27 -07:00
d4f4ff3c5e
Allow None input to compiled functions ( #2621 )
...
* Allow None input to compiled functions
* Allow None input to compiled functions
2025-09-25 08:42:23 -07:00
fbbf3b9b3e
Support pickling array for bfloat16 ( #2586 )
...
* add bfloat16 pickling
* Improvements
* improve
---------
Co-authored-by: Chen-Chen Yeh <ge96noj@mytum.de >
2025-09-22 20:12:15 -07:00
50cc09887f
expose depends ( #2606 )
2025-09-18 10:06:15 -07:00
Cheng