zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-15 14:26:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
1,356 Commits 40 Branches 69 Tags 1.7 GiB
185a879b05
Commit Graph

1356 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Awni Hannun
185a879b05 docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
3df9dab38c docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
b86da5711b docs 2025-08-07 07:46:32 +00:00
Awni Hannun
2988d6c14d docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
680a57bb33 docs update 2025-08-07 07:46:32 +00:00
Awni Hannun
b70128ce73 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
1364ef2259 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
2c0a1a3952 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
aea164fab8 use proper version 2025-08-07 07:46:31 +00:00
Awni Hannun
9abc99cfe9 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
4490e181db docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
e04f658611 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
b04ef9ec37 docs update 2025-08-07 07:46:31 +00:00
Awni Hannun
f2365d5b12 remove uneeded files in docs 2025-08-07 07:46:30 +00:00
Awni Hannun
eea1a5f6ec update docs 2025-08-07 07:46:30 +00:00
Awni Hannun
1c7ebac536 docs update 2025-08-07 07:46:30 +00:00
Awni Hannun
c07de29458 docs up 2025-08-07 07:46:30 +00:00
Awni Hannun
c0dc02b8f4 docs up 2025-08-07 07:46:30 +00:00
Awni Hannun
e9e7ff09dd docs update 2025-08-07 07:46:30 +00:00
Awni Hannun
b024ed7744 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
beda33e1a9 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
83d006dfdd update docs 2025-08-07 07:46:30 +00:00
Awni Hannun
b906475af3 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
0bb6edbe54 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
b728f1fd66 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
3bf234ff88 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
082e939fc7 docs 2025-08-07 07:46:30 +00:00
Awni Hannun
75b55ffa19 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
b8cf6bd778 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
3e96be74e6 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
312037d0f7 docs 2025-08-07 07:46:29 +00:00
Awni Hannun
56be773610
version (#2470) 2025-08-07 00:36:04 -07:00
Jagrit Digani
a9bdd67baa
Add CUDA sdpa vector (#2468) 2025-08-06 21:40:26 -07:00
Angelos Katharopoulos
f2adb5638d
Fix typo in metal command encoder (#2471) 2025-08-06 16:58:23 -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
7bb96e4249
fix cublas on h100 (#2466) 2025-08-06 06:18:58 -07:00
Awni Hannun
fa89f0b150
faster gather qmm sorted test (#2463) 2025-08-05 06:27:40 -07:00
Awni Hannun
ca973d1e83
fix install tags (#2464) 2025-08-04 20:01:23 -07:00
Cheng
828c5f1137
Use SmallVector for shapes and strides (#2454) 
* Use SmallVector for shapes and strides

* Convert SmallVector to tuple
 2025-08-05 09:41:03 +09:00
Gaétan Lepage
7d86a5c108
Feat: add USE_SYSTEM_FMT CMake option (#2219) 2025-08-04 16:36:11 -07:00
Awni Hannun
0b807893a7
fix wraps compile (#2461) 2025-08-04 16:14:18 -07:00
Awni Hannun
6ad0889c8a
default install cuda on linux (#2462) 2025-08-04 15:33:05 -07:00
Zamderax
737dd6d1ac
Add missing <algorithm> header to jit_compiler.cpp (#2460) 
Fixes compilation error on Linux where std::find_if is used on line 121
but the <algorithm> header was not included. While this might work on
some platforms due to transitive includes, it's not guaranteed by the
C++ standard.

Resolves issue #2459
 2025-08-04 14:00:46 -07:00
Cheng
aaf78f4c6b
Use LRU cache for cuda graph (#2448) 
* Use LRU cache for cuda graph

* Remove unused destructor
 2025-08-02 21:28:57 +09:00
Angelos Katharopoulos
8831064493
Fix arctan2 grads (#2453) 2025-08-01 21:06:04 -07:00
Angelos Katharopoulos
be9bc96da4
[CUDA] Matmul utils initial commit (#2441) 2025-08-01 14:22:25 -07:00
Angelos Katharopoulos
86258f292f
[CUDA] Vectorize generated kernels (#2444) 2025-07-31 18:18:57 -07:00
Cheng
b26d88591c
[CUDA] Save primitive inputs faster (#2449) 
* Add more nvtx loggings

* [CUDA] Saving primitive inputs faster

* Remove unneeded check
 2025-08-01 10:16:06 +09:00
Cheng
86c6a15571
[CUDA] Backward convolution (#2431) 2025-08-01 09:54:05 +09:00