mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-31 11:54:40 +08:00
Commit Graph
Select branches
Hide Pull Requests
ab-nf4-quant
alternative_fence
async_all_reduce
attn-update
checkpoint_module
compile-test
cpp20
cuda-sdpa-vector
dynamic_reshape
fences_must_exit
fft
gemm-tuner
gguf_q4_k
gh-pages
gs16
interrupt_eval
io-dev
jagrit06/cuda-gemm-experiment
main
mat-attn
more_donation
packed-quants
priority-queue-eval
q-sdpa
qmm
quantized-kv-update
raw-sockets-distributed
ring-init
sdpa-test
sdpav-backup
simple-gemm
socket-distributed-layers
socket-distributed-layers-gloo
sockets-distributed
split_logsumexp
steel-refactor
stop-fence-term
trellis-quants
winograd_qupdate
#1
#1000
#1002
#1003
#1006
#1007
#1010
#1011
#1014
#1016
#1018
#1019
#102
#1020
#1028
#1030
#1032
#1034
#1035
#1036
#1037
#1038
#1039
#104
#1043
#1053
#1054
#1058
#1059
#1060
#1061
#1064
#1067
#1070
#1074
#1077
#1079
#108
#1081
#1085
#1087
#109
#1091
#1092
#1093
#1097
#1098
#1099
#11
#110
#1100
#1101
#1102
#1104
#1105
#1109
#111
#1110
#1111
#1112
#1113
#1114
#1115
#1116
#1117
#1118
#1119
#1120
#1122
#1123
#1124
#1125
#1126
#1129
#1131
#1132
#1135
#1136
#1137
#1138
#1139
#1140
#1141
#1142
#1147
#1149
#115
#1150
#1151
#1152
#1154
#1157
#116
#1161
#1165
#1167
#1168
#1169
#117
#1172
#1174
#1175
#1176
#1177
#1178
#1179
#118
#1180
#1183
#1184
#1185
#1188
#1189
#119
#1190
#1191
#1194
#1195
#1199
#120
#1200
#1202
#1203
#1204
#1205
#1206
#1208
#1209
#121
#1211
#1212
#1215
#1216
#122
#1221
#1222
#1224
#1227
#1228
#123
#1235
#1236
#1237
#1239
#1242
#1243
#1245
#1246
#1247
#1249
#125
#1252
#1253
#1256
#1260
#1262
#1262
#1263
#1264
#1266
#1268
#1269
#1270
#1273
#1274
#1275
#1278
#1279
#128
#1280
#1281
#1282
#1283
#1285
#1287
#1289
#1291
#1297
#1298
#1299
#1300
#1301
#1304
#1305
#1306
#1307
#1309
#131
#1310
#1314
#1315
#1316
#1318
#1319
#1320
#1323
#1325
#1326
#1327
#1329
#133
#1330
#1332
#1333
#1334
#1336
#1337
#1339
#1340
#1343
#1344
#1346
#1347
#1348
#1349
#1350
#1351
#1352
#1353
#1355
#1356
#1358
#1359
#136
#1360
#1361
#1362
#1365
#1366
#1367
#1368
#1369
#137
#1371
#1372
#1373
#1374
#1376
#1379
#138
#1381
#1383
#1384
#1385
#1387
#1389
#139
#1390
#1391
#1394
#1395
#1396
#1397
#1401
#1402
#1403
#1404
#1405
#1407
#1408
#1410
#1412
#1414
#1415
#1416
#1417
#1418
#1419
#142
#1420
#1421
#1425
#143
#1430
#1431
#1434
#1436
#1437
#144
#1440
#1442
#1444
#1445
#1446
#1447
#1449
#145
#1450
#1451
#1452
#1453
#1455
#1456
#1460
#1461
#1462
#1468
#1470
#1471
#1476
#1477
#1478
#1479
#1482
#1485
#1486
#1488
#149
#1490
#1491
#1492
#1493
#1495
#1496
#1497
#1498
#150
#1501
#1502
#1503
#1506
#1508
#1509
#1510
#1514
#1515
#1515
#1518
#1519
#1521
#1522
#1523
#1524
#1525
#1526
#1528
#1529
#1532
#1534
#1535
#1537
#1539
#1541
#1543
#1545
#1546
#1548
#1550
#1551
#1553
#1555
#1556
#1557
#1558
#156
#1561
#1562
#1563
#1564
#1565
#1566
#1568
#1569
#157
#1570
#1572
#1574
#1575
#1577
#1578
#1579
#158
#1584
#1587
#1589
#159
#1590
#1591
#1594
#1595
#1596
#1597
#1600
#1601
#1603
#1606
#1607
#1609
#161
#1610
#1612
#1613
#1615
#1616
#1617
#1620
#1625
#1626
#1627
#1628
#1629
#1630
#1632
#1634
#1635
#1637
#1638
#1639
#1640
#1642
#1644
#1645
#1646
#1650
#1651
#1652
#1653
#1654
#1655
#1656
#1657
#1658
#1659
#166
#1660
#1661
#1662
#1663
#1664
#1665
#1666
#1667
#1668
#167
#1671
#1672
#1673
#1674
#1675
#1677
#1678
#1679
#1680
#1681
#1682
#1683
#1684
#1685
#1687
#1688
#1689
#1690
#1691
#1692
#1693
#1694
#1695
#1696
#1697
#1698
#1699
#170
#1700
#1701
#1702
#1704
#1705
#1706
#1708
#1709
#1710
#1714
#1715
#1716
#1718
#1719
#1721
#1722
#1723
#1724
#1726
#1727
#1728
#1731
#1732
#1733
#1735
#1736
#1737
#1738
#1740
#1741
#1742
#1743
#1745
#1746
#1747
#1748
#1749
#1750
#1752
#1753
#1754
#1755
#1756
#1757
#1758
#1759
#1760
#1761
#1762
#1763
#1764
#1765
#1768
#1772
#1773
#1774
#1775
#1776
#1777
#1782
#1783
#1784
#1788
#1789
#1789
#1793
#1795
#1797
#1798
#1799
#1801
#1802
#1803
#1805
#1806
#181
#1810
#1811
#1812
#1813
#1814
#1816
#1817
#1819
#1820
#1822
#1825
#1827
#1829
#183
#1830
#1831
#1833
#1834
#1835
#1836
#1837
#1838
#184
#1840
#1843
#1844
#1845
#1848
#185
#1852
#1854
#1856
#1857
#1858
#1859
#186
#1860
#1861
#1862
#1863
#1864
#1865
#1866
#1867
#1869
#187
#1870
#1874
#1875
#1876
#1879
#1882
#1883
#1884
#1885
#1887
#1889
#189
#1890
#1892
#1894
#1896
#1897
#1898
#1899
#190
#1900
#1901
#1902
#1904
#1906
#1911
#1913
#1914
#1915
#1916
#1917
#1920
#1921
#1922
#1923
#1924
#1925
#1926
#1928
#1929
#1931
#1932
#1935
#1936
#1937
#1938
#1939
#1940
#1943
#1944
#1948
#1949
#195
#1950
#1952
#1953
#1955
#1957
#196
#1961
#1962
#1966
#1968
#1969
#1970
#1970
#1972
#1973
#1974
#1975
#1976
#1978
#198
#1980
#1981
#1982
#1983
#1985
#1986
#1987
#1988
#1989
#199
#1990
#1991
#1992
#1995
#1996
#1997
#1998
#1999
#2
#2000
#2004
#2005
#2006
#2007
#2008
#2009
#2011
#2012
#2013
#2014
#2016
#2017
#2018
#202
#2020
#2021
#2024
#2025
#2026
#2027
#2028
#2029
#203
#2031
#2032
#2033
#2035
#2036
#2040
#2041
#2042
#2043
#2044
#2045
#2046
#2047
#2048
#2049
#205
#2051
#2052
#2053
#2054
#2055
#2058
#2059
#2060
#2061
#2062
#2065
#2066
#2068
#2069
#207
#2070
#2071
#2072
#2073
#2074
#2074
#2075
#2078
#2079
#2080
#2081
#2082
#2087
#209
#2090
#2091
#2092
#2094
#2095
#210
#2100
#2101
#2102
#2104
#2104
#2109
#211
#2110
#2114
#2117
#2119
#2121
#2123
#2128
#2129
#2131
#2135
#2136
#2138
#2139
#2141
#2142
#2143
#2145
#2147
#2148
#215
#2150
#2151
#2152
#2153
#2156
#2156
#2157
#2158
#2160
#2161
#2162
#217
#2172
#2173
#2176
#2177
#2178
#2179
#2181
#2182
#2183
#2187
#2188
#2189
#2191
#2192
#2193
#2195
#22
#2201
#2202
#2204
#2206
#2207
#2209
#221
#2210
#2213
#2214
#2215
#2216
#2217
#2219
#222
#2220
#2221
#2223
#2225
#2226
#2230
#2231
#2232
#2234
#2237
#2239
#224
#2240
#2241
#2242
#2243
#2244
#2246
#2247
#2248
#225
#2250
#2255
#2256
#2258
#2259
#226
#2260
#2261
#2262
#2263
#2264
#2265
#2268
#2269
#227
#2270
#2271
#2272
#2274
#2275
#2276
#2277
#228
#2280
#2282
#2283
#2284
#2286
#2287
#2288
#2289
#2290
#2291
#2293
#2294
#2295
#2296
#2297
#2297
#2298
#2299
#230
#2300
#2300
#2302
#2303
#2304
#2306
#2307
#2308
#2311
#2314
#2316
#2317
#2318
#232
#2320
#2321
#2322
#2323
#2324
#2325
#2326
#2327
#2328
#2329
#233
#2330
#2331
#2335
#2336
#2339
#2340
#2341
#2342
#2343
#2345
#2346
#2347
#235
#2350
#2351
#2352
#2354
#2355
#2356
#2357
#2360
#2361
#2362
#2363
#2364
#2365
#2367
#2368
#2370
#2371
#2372
#2375
#2377
#2378
#2379
#2380
#2382
#2383
#2385
#2386
#2387
#2388
#2389
#2392
#2396
#2397
#2398
#2399
#240
#2400
#2401
#2401
#2404
#2406
#2407
#2408
#2409
#2411
#2412
#2413
#2414
#2415
#2416
#2417
#2419
#2420
#2421
#2423
#2424
#2425
#2426
#2427
#2429
#2430
#2431
#2432
#2433
#2434
#2435
#2437
#2438
#2439
#244
#2440
#2441
#2442
#2443
#2444
#2445
#2446
#2447
#2448
#2449
#245
#2450
#2453
#2454
#2455
#2460
#2461
#2462
#2463
#2464
#2465
#2466
#2468
#247
#2470
#2471
#2472
#2473
#2474
#2476
#2477
#2480
#2482
#2483
#2484
#2485
#2486
#2487
#2488
#2489
#249
#2491
#2493
#2494
#2495
#2496
#2499
#250
#2502
#2505
#2506
#2508
#2510
#2511
#2513
#2514
#2515
#2517
#2518
#252
#2520
#2521
#2523
#2524
#2525
#2526
#2527
#2528
#2530
#2531
#2532
#2533
#2534
#2535
#2539
#254
#2541
#2542
#2543
#2544
#2545
#2546
#2548
#2549
#255
#2550
#2551
#2552
#2553
#2554
#2555
#256
#26
#260
#261
#263
#268
#274
#275
#276
#277
#278
#280
#281
#284
#289
#29
#292
#295
#298
#299
#3
#302
#304
#305
#306
#307
#308
#309
#310
#311
#312
#313
#315
#316
#317
#318
#319
#32
#323
#324
#325
#327
#329
#330
#332
#334
#335
#336
#337
#339
#34
#340
#342
#344
#345
#347
#348
#349
#350
#352
#353
#354
#355
#356
#357
#358
#359
#36
#364
#366
#370
#371
#373
#374
#375
#379
#38
#380
#381
#382
#383
#384
#385
#386
#388
#389
#390
#391
#392
#394
#395
#397
#398
#4
#401
#405
#406
#409
#411
#412
#414
#415
#418
#419
#421
#423
#424
#425
#426
#427
#428
#430
#431
#432
#433
#435
#438
#441
#443
#444
#445
#446
#447
#448
#449
#453
#455
#456
#457
#458
#461
#462
#463
#47
#470
#472
#473
#475
#476
#477
#478
#479
#480
#481
#482
#484
#489
#490
#492
#494
#497
#498
#5
#50
#500
#501
#505
#507
#508
#510
#511
#512
#513
#514
#517
#519
#520
#521
#523
#524
#525
#526
#527
#53
#533
#536
#537
#539
#541
#543
#55
#552
#554
#558
#559
#56
#560
#561
#562
#563
#564
#565
#569
#571
#579
#58
#581
#584
#588
#59
#591
#592
#595
#599
#6
#601
#602
#603
#604
#607
#608
#612
#613
#614
#616
#619
#62
#620
#622
#623
#624
#625
#626
#627
#629
#630
#631
#634
#635
#637
#638
#639
#64
#641
#643
#645
#647
#648
#651
#653
#656
#658
#659
#660
#661
#662
#663
#664
#667
#670
#674
#675
#676
#677
#678
#679
#68
#681
#682
#683
#684
#685
#686
#687
#688
#689
#69
#691
#692
#694
#696
#697
#698
#699
#7
#70
#702
#703
#704
#705
#706
#707
#708
#709
#710
#711
#713
#715
#716
#717
#718
#72
#721
#723
#727
#729
#730
#735
#737
#739
#74
#740
#744
#745
#747
#749
#75
#752
#759
#760
#761
#763
#764
#766
#768
#771
#776
#777
#778
#78
#78
#782
#783
#784
#785
#786
#787
#788
#79
#790
#791
#792
#793
#794
#796
#797
#8
#80
#800
#801
#802
#803
#804
#805
#806
#807
#809
#81
#812
#813
#816
#818
#819
#820
#821
#822
#823
#824
#826
#828
#829
#831
#835
#836
#838
#839
#84
#841
#843
#844
#848
#849
#85
#850
#852
#853
#858
#859
#861
#862
#863
#864
#867
#869
#87
#870
#871
#872
#874
#875
#876
#877
#879
#88
#880
#881
#883
#886
#889
#89
#890
#891
#892
#893
#894
#895
#897
#899
#9
#90
#901
#902
#903
#904
#905
#906
#907
#91
#911
#915
#916
#917
#919
#92
#920
#921
#923
#924
#925
#926
#929
#932
#933
#934
#94
#940
#941
#942
#944
#947
#948
#949
#950
#951
#952
#953
#955
#956
#957
#958
#960
#961
#962
#964
#967
#969
#970
#971
#972
#973
#974
#975
#976
#977
#978
#979
#98
#980
#981
#982
#984
#986
#987
#988
#989
#99
#991
#992
#993
#994
#998
v0.0.10
v0.0.11
v0.0.2
v0.0.3
v0.0.4
v0.0.5
v0.0.6
v0.0.7
v0.0.9
v0.1.0
v0.10.0
v0.11.0
v0.11.1
v0.12.0
v0.12.1
v0.12.2
v0.13.0
v0.13.1
v0.14.0
v0.14.1
v0.15.0
v0.15.1
v0.15.2
v0.16.0
v0.16.1
v0.16.2
v0.16.3
v0.17.0
v0.17.1
v0.17.2
v0.17.3
v0.18.0
v0.18.1
v0.19.0
v0.19.1
v0.19.2
v0.19.3
v0.2.0
v0.20.0
v0.21.0
v0.21.1
v0.22.0
v0.22.1
v0.23.0
v0.23.1
v0.23.2
v0.24.0
v0.24.1
v0.24.2
v0.25.0
v0.25.1
v0.25.2
v0.26.0
v0.26.1
v0.26.2
v0.26.3
v0.26.5
v0.27.1
v0.28.0
v0.29.0
v0.3.0
v0.4.0
v0.5.0
v0.5.1
v0.6.0
v0.7.0
v0.8.0
v0.8.1
v0.9.0
v0.9.1
Select branches
Hide Pull Requests
ab-nf4-quant
alternative_fence
async_all_reduce
attn-update
checkpoint_module
compile-test
cpp20
cuda-sdpa-vector
dynamic_reshape
fences_must_exit
fft
gemm-tuner
gguf_q4_k
gh-pages
gs16
interrupt_eval
io-dev
jagrit06/cuda-gemm-experiment
main
mat-attn
more_donation
packed-quants
priority-queue-eval
q-sdpa
qmm
quantized-kv-update
raw-sockets-distributed
ring-init
sdpa-test
sdpav-backup
simple-gemm
socket-distributed-layers
socket-distributed-layers-gloo
sockets-distributed
split_logsumexp
steel-refactor
stop-fence-term
trellis-quants
winograd_qupdate
#1
#1000
#1002
#1003
#1006
#1007
#1010
#1011
#1014
#1016
#1018
#1019
#102
#1020
#1028
#1030
#1032
#1034
#1035
#1036
#1037
#1038
#1039
#104
#1043
#1053
#1054
#1058
#1059
#1060
#1061
#1064
#1067
#1070
#1074
#1077
#1079
#108
#1081
#1085
#1087
#109
#1091
#1092
#1093
#1097
#1098
#1099
#11
#110
#1100
#1101
#1102
#1104
#1105
#1109
#111
#1110
#1111
#1112
#1113
#1114
#1115
#1116
#1117
#1118
#1119
#1120
#1122
#1123
#1124
#1125
#1126
#1129
#1131
#1132
#1135
#1136
#1137
#1138
#1139
#1140
#1141
#1142
#1147
#1149
#115
#1150
#1151
#1152
#1154
#1157
#116
#1161
#1165
#1167
#1168
#1169
#117
#1172
#1174
#1175
#1176
#1177
#1178
#1179
#118
#1180
#1183
#1184
#1185
#1188
#1189
#119
#1190
#1191
#1194
#1195
#1199
#120
#1200
#1202
#1203
#1204
#1205
#1206
#1208
#1209
#121
#1211
#1212
#1215
#1216
#122
#1221
#1222
#1224
#1227
#1228
#123
#1235
#1236
#1237
#1239
#1242
#1243
#1245
#1246
#1247
#1249
#125
#1252
#1253
#1256
#1260
#1262
#1262
#1263
#1264
#1266
#1268
#1269
#1270
#1273
#1274
#1275
#1278
#1279
#128
#1280
#1281
#1282
#1283
#1285
#1287
#1289
#1291
#1297
#1298
#1299
#1300
#1301
#1304
#1305
#1306
#1307
#1309
#131
#1310
#1314
#1315
#1316
#1318
#1319
#1320
#1323
#1325
#1326
#1327
#1329
#133
#1330
#1332
#1333
#1334
#1336
#1337
#1339
#1340
#1343
#1344
#1346
#1347
#1348
#1349
#1350
#1351
#1352
#1353
#1355
#1356
#1358
#1359
#136
#1360
#1361
#1362
#1365
#1366
#1367
#1368
#1369
#137
#1371
#1372
#1373
#1374
#1376
#1379
#138
#1381
#1383
#1384
#1385
#1387
#1389
#139
#1390
#1391
#1394
#1395
#1396
#1397
#1401
#1402
#1403
#1404
#1405
#1407
#1408
#1410
#1412
#1414
#1415
#1416
#1417
#1418
#1419
#142
#1420
#1421
#1425
#143
#1430
#1431
#1434
#1436
#1437
#144
#1440
#1442
#1444
#1445
#1446
#1447
#1449
#145
#1450
#1451
#1452
#1453
#1455
#1456
#1460
#1461
#1462
#1468
#1470
#1471
#1476
#1477
#1478
#1479
#1482
#1485
#1486
#1488
#149
#1490
#1491
#1492
#1493
#1495
#1496
#1497
#1498
#150
#1501
#1502
#1503
#1506
#1508
#1509
#1510
#1514
#1515
#1515
#1518
#1519
#1521
#1522
#1523
#1524
#1525
#1526
#1528
#1529
#1532
#1534
#1535
#1537
#1539
#1541
#1543
#1545
#1546
#1548
#1550
#1551
#1553
#1555
#1556
#1557
#1558
#156
#1561
#1562
#1563
#1564
#1565
#1566
#1568
#1569
#157
#1570
#1572
#1574
#1575
#1577
#1578
#1579
#158
#1584
#1587
#1589
#159
#1590
#1591
#1594
#1595
#1596
#1597
#1600
#1601
#1603
#1606
#1607
#1609
#161
#1610
#1612
#1613
#1615
#1616
#1617
#1620
#1625
#1626
#1627
#1628
#1629
#1630
#1632
#1634
#1635
#1637
#1638
#1639
#1640
#1642
#1644
#1645
#1646
#1650
#1651
#1652
#1653
#1654
#1655
#1656
#1657
#1658
#1659
#166
#1660
#1661
#1662
#1663
#1664
#1665
#1666
#1667
#1668
#167
#1671
#1672
#1673
#1674
#1675
#1677
#1678
#1679
#1680
#1681
#1682
#1683
#1684
#1685
#1687
#1688
#1689
#1690
#1691
#1692
#1693
#1694
#1695
#1696
#1697
#1698
#1699
#170
#1700
#1701
#1702
#1704
#1705
#1706
#1708
#1709
#1710
#1714
#1715
#1716
#1718
#1719
#1721
#1722
#1723
#1724
#1726
#1727
#1728
#1731
#1732
#1733
#1735
#1736
#1737
#1738
#1740
#1741
#1742
#1743
#1745
#1746
#1747
#1748
#1749
#1750
#1752
#1753
#1754
#1755
#1756
#1757
#1758
#1759
#1760
#1761
#1762
#1763
#1764
#1765
#1768
#1772
#1773
#1774
#1775
#1776
#1777
#1782
#1783
#1784
#1788
#1789
#1789
#1793
#1795
#1797
#1798
#1799
#1801
#1802
#1803
#1805
#1806
#181
#1810
#1811
#1812
#1813
#1814
#1816
#1817
#1819
#1820
#1822
#1825
#1827
#1829
#183
#1830
#1831
#1833
#1834
#1835
#1836
#1837
#1838
#184
#1840
#1843
#1844
#1845
#1848
#185
#1852
#1854
#1856
#1857
#1858
#1859
#186
#1860
#1861
#1862
#1863
#1864
#1865
#1866
#1867
#1869
#187
#1870
#1874
#1875
#1876
#1879
#1882
#1883
#1884
#1885
#1887
#1889
#189
#1890
#1892
#1894
#1896
#1897
#1898
#1899
#190
#1900
#1901
#1902
#1904
#1906
#1911
#1913
#1914
#1915
#1916
#1917
#1920
#1921
#1922
#1923
#1924
#1925
#1926
#1928
#1929
#1931
#1932
#1935
#1936
#1937
#1938
#1939
#1940
#1943
#1944
#1948
#1949
#195
#1950
#1952
#1953
#1955
#1957
#196
#1961
#1962
#1966
#1968
#1969
#1970
#1970
#1972
#1973
#1974
#1975
#1976
#1978
#198
#1980
#1981
#1982
#1983
#1985
#1986
#1987
#1988
#1989
#199
#1990
#1991
#1992
#1995
#1996
#1997
#1998
#1999
#2
#2000
#2004
#2005
#2006
#2007
#2008
#2009
#2011
#2012
#2013
#2014
#2016
#2017
#2018
#202
#2020
#2021
#2024
#2025
#2026
#2027
#2028
#2029
#203
#2031
#2032
#2033
#2035
#2036
#2040
#2041
#2042
#2043
#2044
#2045
#2046
#2047
#2048
#2049
#205
#2051
#2052
#2053
#2054
#2055
#2058
#2059
#2060
#2061
#2062
#2065
#2066
#2068
#2069
#207
#2070
#2071
#2072
#2073
#2074
#2074
#2075
#2078
#2079
#2080
#2081
#2082
#2087
#209
#2090
#2091
#2092
#2094
#2095
#210
#2100
#2101
#2102
#2104
#2104
#2109
#211
#2110
#2114
#2117
#2119
#2121
#2123
#2128
#2129
#2131
#2135
#2136
#2138
#2139
#2141
#2142
#2143
#2145
#2147
#2148
#215
#2150
#2151
#2152
#2153
#2156
#2156
#2157
#2158
#2160
#2161
#2162
#217
#2172
#2173
#2176
#2177
#2178
#2179
#2181
#2182
#2183
#2187
#2188
#2189
#2191
#2192
#2193
#2195
#22
#2201
#2202
#2204
#2206
#2207
#2209
#221
#2210
#2213
#2214
#2215
#2216
#2217
#2219
#222
#2220
#2221
#2223
#2225
#2226
#2230
#2231
#2232
#2234
#2237
#2239
#224
#2240
#2241
#2242
#2243
#2244
#2246
#2247
#2248
#225
#2250
#2255
#2256
#2258
#2259
#226
#2260
#2261
#2262
#2263
#2264
#2265
#2268
#2269
#227
#2270
#2271
#2272
#2274
#2275
#2276
#2277
#228
#2280
#2282
#2283
#2284
#2286
#2287
#2288
#2289
#2290
#2291
#2293
#2294
#2295
#2296
#2297
#2297
#2298
#2299
#230
#2300
#2300
#2302
#2303
#2304
#2306
#2307
#2308
#2311
#2314
#2316
#2317
#2318
#232
#2320
#2321
#2322
#2323
#2324
#2325
#2326
#2327
#2328
#2329
#233
#2330
#2331
#2335
#2336
#2339
#2340
#2341
#2342
#2343
#2345
#2346
#2347
#235
#2350
#2351
#2352
#2354
#2355
#2356
#2357
#2360
#2361
#2362
#2363
#2364
#2365
#2367
#2368
#2370
#2371
#2372
#2375
#2377
#2378
#2379
#2380
#2382
#2383
#2385
#2386
#2387
#2388
#2389
#2392
#2396
#2397
#2398
#2399
#240
#2400
#2401
#2401
#2404
#2406
#2407
#2408
#2409
#2411
#2412
#2413
#2414
#2415
#2416
#2417
#2419
#2420
#2421
#2423
#2424
#2425
#2426
#2427
#2429
#2430
#2431
#2432
#2433
#2434
#2435
#2437
#2438
#2439
#244
#2440
#2441
#2442
#2443
#2444
#2445
#2446
#2447
#2448
#2449
#245
#2450
#2453
#2454
#2455
#2460
#2461
#2462
#2463
#2464
#2465
#2466
#2468
#247
#2470
#2471
#2472
#2473
#2474
#2476
#2477
#2480
#2482
#2483
#2484
#2485
#2486
#2487
#2488
#2489
#249
#2491
#2493
#2494
#2495
#2496
#2499
#250
#2502
#2505
#2506
#2508
#2510
#2511
#2513
#2514
#2515
#2517
#2518
#252
#2520
#2521
#2523
#2524
#2525
#2526
#2527
#2528
#2530
#2531
#2532
#2533
#2534
#2535
#2539
#254
#2541
#2542
#2543
#2544
#2545
#2546
#2548
#2549
#255
#2550
#2551
#2552
#2553
#2554
#2555
#256
#26
#260
#261
#263
#268
#274
#275
#276
#277
#278
#280
#281
#284
#289
#29
#292
#295
#298
#299
#3
#302
#304
#305
#306
#307
#308
#309
#310
#311
#312
#313
#315
#316
#317
#318
#319
#32
#323
#324
#325
#327
#329
#330
#332
#334
#335
#336
#337
#339
#34
#340
#342
#344
#345
#347
#348
#349
#350
#352
#353
#354
#355
#356
#357
#358
#359
#36
#364
#366
#370
#371
#373
#374
#375
#379
#38
#380
#381
#382
#383
#384
#385
#386
#388
#389
#390
#391
#392
#394
#395
#397
#398
#4
#401
#405
#406
#409
#411
#412
#414
#415
#418
#419
#421
#423
#424
#425
#426
#427
#428
#430
#431
#432
#433
#435
#438
#441
#443
#444
#445
#446
#447
#448
#449
#453
#455
#456
#457
#458
#461
#462
#463
#47
#470
#472
#473
#475
#476
#477
#478
#479
#480
#481
#482
#484
#489
#490
#492
#494
#497
#498
#5
#50
#500
#501
#505
#507
#508
#510
#511
#512
#513
#514
#517
#519
#520
#521
#523
#524
#525
#526
#527
#53
#533
#536
#537
#539
#541
#543
#55
#552
#554
#558
#559
#56
#560
#561
#562
#563
#564
#565
#569
#571
#579
#58
#581
#584
#588
#59
#591
#592
#595
#599
#6
#601
#602
#603
#604
#607
#608
#612
#613
#614
#616
#619
#62
#620
#622
#623
#624
#625
#626
#627
#629
#630
#631
#634
#635
#637
#638
#639
#64
#641
#643
#645
#647
#648
#651
#653
#656
#658
#659
#660
#661
#662
#663
#664
#667
#670
#674
#675
#676
#677
#678
#679
#68
#681
#682
#683
#684
#685
#686
#687
#688
#689
#69
#691
#692
#694
#696
#697
#698
#699
#7
#70
#702
#703
#704
#705
#706
#707
#708
#709
#710
#711
#713
#715
#716
#717
#718
#72
#721
#723
#727
#729
#730
#735
#737
#739
#74
#740
#744
#745
#747
#749
#75
#752
#759
#760
#761
#763
#764
#766
#768
#771
#776
#777
#778
#78
#78
#782
#783
#784
#785
#786
#787
#788
#79
#790
#791
#792
#793
#794
#796
#797
#8
#80
#800
#801
#802
#803
#804
#805
#806
#807
#809
#81
#812
#813
#816
#818
#819
#820
#821
#822
#823
#824
#826
#828
#829
#831
#835
#836
#838
#839
#84
#841
#843
#844
#848
#849
#85
#850
#852
#853
#858
#859
#861
#862
#863
#864
#867
#869
#87
#870
#871
#872
#874
#875
#876
#877
#879
#88
#880
#881
#883
#886
#889
#89
#890
#891
#892
#893
#894
#895
#897
#899
#9
#90
#901
#902
#903
#904
#905
#906
#907
#91
#911
#915
#916
#917
#919
#92
#920
#921
#923
#924
#925
#926
#929
#932
#933
#934
#94
#940
#941
#942
#944
#947
#948
#949
#950
#951
#952
#953
#955
#956
#957
#958
#960
#961
#962
#964
#967
#969
#970
#971
#972
#973
#974
#975
#976
#977
#978
#979
#98
#980
#981
#982
#984
#986
#987
#988
#989
#99
#991
#992
#993
#994
#998
v0.0.10
v0.0.11
v0.0.2
v0.0.3
v0.0.4
v0.0.5
v0.0.6
v0.0.7
v0.0.9
v0.1.0
v0.10.0
v0.11.0
v0.11.1
v0.12.0
v0.12.1
v0.12.2
v0.13.0
v0.13.1
v0.14.0
v0.14.1
v0.15.0
v0.15.1
v0.15.2
v0.16.0
v0.16.1
v0.16.2
v0.16.3
v0.17.0
v0.17.1
v0.17.2
v0.17.3
v0.18.0
v0.18.1
v0.19.0
v0.19.1
v0.19.2
v0.19.3
v0.2.0
v0.20.0
v0.21.0
v0.21.1
v0.22.0
v0.22.1
v0.23.0
v0.23.1
v0.23.2
v0.24.0
v0.24.1
v0.24.2
v0.25.0
v0.25.1
v0.25.2
v0.26.0
v0.26.1
v0.26.2
v0.26.3
v0.26.5
v0.27.1
v0.28.0
v0.29.0
v0.3.0
v0.4.0
v0.5.0
v0.5.1
v0.6.0
v0.7.0
v0.8.0
v0.8.1
v0.9.0
v0.9.1
-
6b6e1d3ac4
fix custom kernel test
Awni Hannun
2025-08-17 06:11:46 -0700 -
f433c9a421
Revert back to old rms norm kernel
Cheng
2025-08-13 04:13:26 -0700 -
3a2b90fc1a
Implement forward rms_norm with cuDNN
Cheng
2025-08-12 04:42:58 -0700 -
290b45eba3
Add RAII managed CudaGraph class
Cheng
2025-08-11 18:20:00 -0700 -
c422050ca7
Cheng
2025-08-17 19:13:01 +0900 -
753081052e
Ensure no oob read in gemv_masked
Angelos Katharopoulos
2025-08-17 00:40:32 -0700 -
b33457ae4d
Ensure small sort doesn't use indices if not argsort
Angelos Katharopoulos
2025-08-16 21:43:24 -0700 -
ebfc6777dd
Update cuDNN Frontend to v1.14
Cheng
2025-08-16 17:18:24 -0700 -
708ca546b2
use defaults for scalar ops
Awni Hannun
2025-08-16 06:36:46 -0700 -
1ba18ff7d9
Cheng
2025-08-16 10:09:18 +0900 -
37b440faa8
Cheng
2025-08-16 09:01:10 +0900 -
888b13ed63
Cheng
2025-08-16 08:17:24 +0900 -
4abb218d21
Cheng
2025-08-16 07:57:30 +0900 -
6441c21a94
Awni Hannun
2025-08-15 15:04:12 -0700 -
bd9977acbb
copy general
Awni Hannun
2025-08-15 13:38:56 -0700 -
102f3ba579
binary two
Awni Hannun
2025-08-15 12:36:44 -0700 -
5e542d98e0
fix + comment
Awni Hannun
2025-08-15 12:11:23 -0700 -
f403ea1764
faster general ops + reorg
Awni Hannun
2025-08-15 10:59:34 -0700 -
1034009b82
faster general unary op
Awni Hannun
2025-08-06 19:57:37 -0700 -
f852acdeed
The naive_conv_2d is no longer used
Cheng
2025-08-15 10:29:53 +0900 -
400f8457ea
Experimenting with a gemm based on the cuda steel utils
jagrit06/cuda-gemm-experiment
Jagrit Digani
2025-08-14 11:27:50 -0700 -
adb64ea409
Put the reshape utils in gpu/copy.h
Cheng
2025-08-14 03:31:12 -0700 -
5b05aaad95
[CUDA] Fix conv grads with groups
Cheng
2025-08-14 03:21:00 -0700 -
8bf8034ffd
Put reshape utils in one file
Cheng
2025-08-14 16:59:38 +0900 -
d00175807f
Remove the hack around SmallVector in cpu compile
Cheng
2025-08-14 10:58:08 +0900 -
e0ed742ef2
Clean up code handling both std::vector and SmallVector
Cheng
2025-08-14 09:07:45 +0900 -
dfb5022eab
Cheng
2025-08-13 09:37:40 +0900 -
ac207ce7aa
Daniel Yeh
2025-08-12 21:29:02 +0200 -
3ebc047168
Add RAII managed CudaGraph class
Cheng
2025-08-11 18:20:00 -0700 -
3ec3eadf5d
Rename cu::Matmul to CublasGemm
Cheng
2025-08-11 08:49:38 +0900 -
fce53b61d6
Abe Leininger
2025-08-12 02:05:33 -0500 -
8ae4a76308
Angelos Katharopoulos
2025-08-12 00:03:42 -0700 -
f2073a0d46
Use CMake <4.1 to avoid the nvpl error
Angelos Katharopoulos
2025-08-11 23:15:05 -0700 -
c9e8cc2856
Add acknowledgment for adaptive max pooling contribution
Vincent Amato
2025-08-11 23:49:37 -0400 -
68c7be55bb
Add AdaptiveMaxPool1d, AdaptiveMaxPool2d, and AdaptiveMaxPool3d layers
Vincent Amato
2025-08-11 23:45:52 -0400 -
c59b46a488
Restyled pooling
Vincent Amato
2025-08-11 23:15:38 -0400 -
652a143b64
Update ACKNOWLEDGMENTS.md to include AdaptiveAvgPool1d
Vincent Amato
2025-08-11 23:13:38 -0400 -
e4530007ae
Add AdaptiveAvgPool1d layer
Vincent Amato
2025-08-11 23:10:56 -0400 -
c320e72635
Add AdaptiveAvgPool2d and AdaptiveAvgPool3d to docs
Vincent Amato
2025-08-11 22:34:58 -0400 -
89b3f69a56
Refactor adaptive pooling for style consistency
Vincent Amato
2025-08-11 22:12:14 -0400 -
634ce07a3e
Add AdaptiveAvgPool2d and AdaptiveAvgPool3d
Vincent Amato
2025-08-11 21:17:49 -0400 -
e9272203a2
Add Vincent Amato to ACKNOWLEDGMENTS.md
Vincent Amato
2025-08-11 20:48:26 -0400 -
1d9ce9d744
Add cyclic_lr scheduler
Vincent Amato
2025-08-11 20:40:33 -0400 -
b7e5e64b13
Add Vincent Amato to ACKNOWLEDGMENTS.md
Vincent Amato
2025-08-11 19:42:01 -0400 -
d74a34d2fc
Add cosine_annealing_warm_restarts to scheduler documentation
Vincent Amato
2025-08-11 19:25:04 -0400 -
84ef89f548
Add CosineAnnealingWarmRestarts scheduler
Vincent Amato
2025-08-11 19:17:47 -0400 -
02d6079626
make code blocks copyable
Chen-Chen Yeh
2025-08-09 18:39:47 +0200 -
984cefb14d
CUDA_VISIBLE_DEVICES to local rank
Anastasiia Filippova
2025-08-09 01:43:14 +0200 -
7fde1b6a1e
Cheng
2025-08-09 06:07:17 +0900 -
c28cff7763
fix reduce sum/prod overflow
aleinin
2025-08-08 03:40:25 -0500 -
79f46f1146
Fix logsumexp/softmax not fused for some cases
Cheng
2025-08-07 15:55:22 +0900 -
aa7b47481a
Cheng
2025-08-08 15:23:30 +0900 -
dadf8d9c93
repeat host -> proc per node
Anastasiia Filippova
2025-08-07 15:09:46 +0200 -
389276e2b8
typo
Anastasiia Filippova
2025-08-07 14:16:34 +0200 -
2e255c8eb4
fixed typo
Anastasiia Filippova
2025-08-07 14:02:38 +0200 -
062aa80b84
minor changer to mlx.launch
Anastasiia Filippova
2025-08-07 13:20:55 +0200 -
f540b1d612
nccl backend
Anastasiia Filippova
2025-08-07 13:11:56 +0200 -
56be773610
Awni Hannun
2025-08-07 00:36:04 -0700 -
a9bdd67baa
Jagrit Digani
2025-08-06 21:40:26 -0700 -
7fa520e955
Remove batch sdpa
Angelos Katharopoulos
2025-08-06 20:26:01 -0700 -
a22d0bf273
Add stricter condition to matrix sdpa
sdpav-backup
Angelos Katharopoulos
2025-08-06 19:51:14 -0700 -
b18f80164b
[CUDA] Optimize set_mm_device_pointers for small ndim
Cheng
2025-08-06 19:23:58 -0700 -
f2adb5638d
Angelos Katharopoulos
2025-08-06 16:58:23 -0700 -
b684d76e8e
Fix typo in metal command encoder
Angelos Katharopoulos
2025-08-06 16:30:02 -0700 -
f3787b84ba
version
Awni Hannun
2025-08-06 15:49:03 -0700 -
728d4db582
Luca Vivona
2025-08-06 18:34:59 -0400 -
99d8de8445
Fix cudnn routing
Jagrit Digani
2025-08-06 15:05:58 -0700 -
c66b76a8c8
Update routing
Jagrit Digani
2025-08-06 15:01:15 -0700 -
f81edd184f
Complete 2 pass sdpav
Jagrit Digani
2025-08-06 13:57:40 -0700 -
7f8ba2a003
[WIP] 2 pass sdpav
Jagrit Digani
2025-08-06 09:54:41 -0700 -
c28249b81a
Add more nvtx range for debug
Jagrit Digani
2025-08-01 12:49:24 -0700 -
e74bcdc5e3
Add sdpa file
Jagrit Digani
2025-07-25 12:30:50 -0700 -
d8ed6c1aa3
Add base cudnn attention support
Jagrit Digani
2025-07-25 12:30:22 -0700 -
db5c7efcf6
Awni Hannun
2025-08-06 06:19:12 -0700 -
7bb96e4249
Awni Hannun
2025-08-06 06:18:58 -0700 -
193dcb8553
fix cublas on h100
Awni Hannun
2025-08-05 20:05:13 -0700 -
d491cc9d8a
revert default cuda install
Awni Hannun
2025-08-05 15:53:49 -0700 -
985b50619b
revert default cuda install
Awni Hannun
2025-08-05 15:52:36 -0700 -
fa89f0b150
Awni Hannun
2025-08-05 06:27:40 -0700 -
8ff54a9595
Simplify the utils a bit
Angelos Katharopoulos
2025-08-04 23:01:36 -0700 -
ca973d1e83
Awni Hannun
2025-08-04 20:01:23 -0700 -
87cefe0905
fix install tags
Awni Hannun
2025-08-04 19:56:51 -0700 -
696c66d3d3
bc6f00c00e
into828c5f1137
Anastasiia Filippova
2025-08-04 20:07:46 -0600 -
828c5f1137
Cheng
2025-08-05 09:41:03 +0900 -
296860d2fa
faster gather qmm sorted test
Awni Hannun
2025-08-04 17:05:36 -0700 -
bc6f00c00e
Changed nccl reduction to be a parrt of cuda grapph
Anastasiia Filippova
2025-08-05 02:00:52 +0200 -
58eed7e0b5
Merge branch 'main' into nccl_backend
Anastasiia Filippova
2025-08-05 01:51:14 +0200 -
7d86a5c108
Gaétan Lepage
2025-08-05 01:36:11 +0200 -
0b807893a7
Awni Hannun
2025-08-04 16:14:18 -0700 -
6ad0889c8a
Awni Hannun
2025-08-04 15:33:05 -0700 -
434ed933c5
default install cuda on linux
Awni Hannun
2025-08-04 14:52:22 -0700 -
737dd6d1ac
Zamderax
2025-08-04 14:00:46 -0700 -
d6d994a385
fix wraps compile
Awni Hannun
2025-08-04 13:47:29 -0700 -
372bffcf1b
Add missing <algorithm> header to jit_compiler.cpp
Zamderax
2025-08-04 11:35:54 -0700 -
8d1181d2b9
Convert SmallVector to tuple
Cheng
2025-08-04 19:29:25 +0900 -
df9ac5f2f9
Add top-level namespace access for gradient control functions
Yannick Muller
2025-08-03 00:46:27 -0400 -
323a8e958d
Format code with pre-commit hooks
Yannick Müller
2025-08-02 23:33:00 -0400 -
1e4bd653db
Implement no_grad functionality following PyTorch API
Yannick Müller
2025-08-02 23:27:37 -0400 -
68e9c60d22
Use SmallVector for shapes and strides
Cheng
2025-08-01 21:38:10 +0900 -
aaf78f4c6b
Cheng
2025-08-02 21:28:57 +0900