 Cheng
|
8ef539522c
|
Fix failing python tests on Windows (#3076)
|
2026-01-30 17:50:18 +09:00 |
|
|
Awni Hannun
|
fcbdd05022
|
More useful error for large indices (#3079)
|
2026-01-29 13:02:39 -08:00 |
|
|
atharva
|
590b4f1c16
|
Fix ALiBi slopes for non-power-of-2 num_heads (#3071)
|
2026-01-29 07:23:11 -08:00 |
|
|
Anri Lombard
|
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 |
|
|
stef
|
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 |
|
|
Awni Hannun
|
4912cc47c2
|
Fp qmv (#2984)
|
2026-01-27 06:33:06 -08:00 |
|
|
Daniel Hiltgen
|
a828e769be
|
GPU discovery (#3055)
Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>
|
2026-01-26 09:54:13 -08:00 |
|
|
Nripesh Niketan
|
b6aa03e5b8
|
Update pre-commit hooks and versions for clang-format, black, and isort (#3059)
|
2026-01-26 06:57:04 -08:00 |
|
|
Awni Hannun
|
5bd99dd5ec
|
Fix flaky macOS test (#3063)
|
2026-01-25 16:40:57 -08:00 |
|
|
Cheng
|
fd27829efa
|
Build and test python package on Windows CI (#3049)
|
2026-01-24 07:22:36 +09:00 |
|
|
Awni Hannun
|
9bac6f8584
|
Allow take on empty array when it makes sense (#3046)
|
2026-01-23 07:25:46 -08:00 |
|
|
Awni Hannun
|
1d56dfdf59
|
Use higher precision for linspace with double (#3029)
|
2026-01-21 06:20:50 -08:00 |
|
|
XXXXRT666
|
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 |
|
|
Cheng
|
d96a2bdf57
|
Fix python package install path in stubgen (#3009)
|
2026-01-19 09:34:02 +09:00 |
|
|
Tarjei Mandt
|
ca14d3d835
|
Fix sharding of quantized models with non-power-of-2 bits (#3006)
|
2026-01-18 07:21:56 -08:00 |
|
|
gufengc
|
d2bef3c6bb
|
fix distributed all_to_sharded bias shard axis from -2 to -1 (#2987)
|
2026-01-17 06:51:42 -08:00 |
|
|
Angelos Katharopoulos
|
3fe7794f22
|
Reverts changing the MLX_IBV_DEVICES to MLX_JACCL_DEVICES (#2999)
|
2026-01-14 15:44:17 -08:00 |
|
|
Awni Hannun
|
2469fc2939
|
patch bump for next release (#2991)
|
2026-01-14 08:46:09 -08:00 |
|
|
Awni Hannun
|
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 |
|
|
MillaFleurs
|
4160ec10f7
|
Fix RandomBits::is_equivalent to include width (#2978)
Co-authored-by: KD2YCU <me@kd2ycu.com>
Co-authored-by: Angelos Katharopoulos <katharas@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>
|
2026-01-13 12:42:37 -08:00 |
|
|
Evan Quiney
|
a8197795f5
|
replace MLX_IBV_COORDINATOR with MLX_JACCL_COORDINATOR (#2986)
|
2026-01-13 11:26:25 -08:00 |
|
|
CCYeh
|
7b1c46982a
|
fix doc (#2988)
|
2026-01-12 13:33:26 -08:00 |
|
|
Anri Lombard
|
edab937248
|
Add asarray to __array_namespace__ (#2966)
|
2026-01-12 06:16:27 -08:00 |
|
|
Anastasiia Filippova
|
503731727d
|
QQ linear (#2931)
|
2026-01-05 11:20:54 -08:00 |
|
|
Awni Hannun
|
1680b6fe38
|
fix numpy dtype bug (#2960)
|
2026-01-05 11:20:40 -08:00 |
|
|
1ndig0
|
1df6c2a009
|
Fix doc issues in mlx.nn.init.he_normal and mlx.nn.hard_tanh (#2968)
Co-authored-by: Awni Hannun <awni@apple.com>
|
2026-01-05 07:23:41 -08:00 |
|
|
Cheng
|
26dfe4f651
|
Fetch nanobind with cmake (#2949)
|
2025-12-24 10:23:45 +09:00 |
|
|
Cheng
|
1d21d0e696
|
[CUDA] Implement gather_mm_rhs (#2902)
|
2025-12-24 09:42:56 +09:00 |
|
|
Awni Hannun
|
1eef1d155c
|
Metal/CPU nvfp4 and mxfp8 (#2946)
|
2025-12-22 20:45:19 -08:00 |
|
|
Angelos Katharopoulos
|
9cfda1a86e
|
Fixes in mlx.distributed_config (#2947)
|
2025-12-22 17:38:52 -08:00 |
|
|
Awni Hannun
|
c0fea26ed2
|
Fix for non row-contig scales (#2941)
|
2025-12-21 06:12:41 -08:00 |
|
|
Angelos Katharopoulos
|
d9f4d8d508
|
Fix pid in local launch (#2936)
|
2025-12-19 13:09:15 -08:00 |
|
|
Cheng
|
fc19a08caa
|
Set install rpath of python bindings with cmake (#2934)
|
2025-12-19 16:43:00 +09:00 |
|
|
Cheng
|
b2e2b19bf7
|
Set rpath with cmake for CUDA build (#2932)
|
2025-12-19 12:53:38 +09:00 |
|
|
Angelos Katharopoulos
|
9194ec20a8
|
Thunderbolt RDMA communications backend (#2808)
|
2025-12-17 11:27:54 -08:00 |
|
|
Anastasiia Filippova
|
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 |
|
|
Awni Hannun
|
81dfe5f137
|
Fix grad in place updates (#2899)
|
2025-12-11 14:44:58 -08:00 |
|
|
Nathan Goldbaum
|
e1fee0074b
|
Update nanobind pin to most recent version (#2896)
|
2025-12-11 06:07:36 -08:00 |
|
|
CCYeh
|
3c8ce9b00e
|
Fix input buffer donation in compile (#2897)
|
2025-12-11 06:07:03 -08:00 |
|
|
Angelos Katharopoulos
|
997cfc7699
|
Add a 2-pass col reduce for CUDA (#2863)
|
2025-12-04 15:53:59 -08:00 |
|
|
Awni Hannun
|
1fa8dc5797
|
Do a PyPi release for cuda on arm (#2866)
|
2025-12-04 15:28:29 -08:00 |
|
|
Awni Hannun
|
a6d6717181
|
fix compile copying (#2871)
|
2025-12-04 12:32:56 -08:00 |
|
|
Awni Hannun
|
941cfe23d7
|
Layer norm throws on dimension mismatch (#2870)
|
2025-12-04 11:21:05 -08:00 |
|
|
romanoneg
|
9abb0b8123
|
Added support for pytree types that inherit from tuple and typing.namedtuple (#2845)
|
2025-12-04 11:06:45 -08:00 |
|
|
Tian En "TianHeng
|
50d3914c67
|
Update gumbel function signature parameters (#2868)
|
2025-12-03 15:37:35 -08:00 |
|
|
Awni Hannun
|
cacbdbf995
|
Fix init from double (#2861)
|
2025-12-03 06:08:11 -08:00 |
|
|
Awni Hannun
|
d8ceae7b77
|
Reduce JVP (#2854)
|
2025-12-02 16:17:47 -08:00 |
|
|
Awni Hannun
|
eff0e31f00
|
Fix export scatters (#2852)
|
2025-12-02 11:24:40 -08:00 |
|
|
CCYeh
|
8879ee00eb
|
Support more Numpy interfaces for masked_scatter (#2832)
|
2025-12-01 17:51:02 -08:00 |
|
|
Chaoran Yu
|
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 |
|