mlx-examples

zhangyiss/mlx-examples

Fork 0

mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-10-31 02:48:07 +08:00

Commit Graph

Select branches

Hide Pull Requests

awq

awq-tq

dist-eval

distributed-layers

flux-dist-improv

flux-qlora

load-gguf

main

openlm

packed-quants

#1

#10

#1001

#1003

#1004

#1006

#1009

#1012

#1013

#1014

#1015

#1016

#102

#1020

#1023

#1024

#1026

#1027

#1028

#103

#1035

#1036

#1037

#1038

#1040

#1045

#1047

#1048

#1048

#1049

#105

#1050

#1054

#1055

#106

#1061

#1062

#1063

#1065

#1068

#1069

#107

#1070

#1072

#1074

#1075

#1078

#1079

#108

#1080

#1081

#1082

#1085

#1089

#1090

#1092

#1093

#1094

#1096

#1099

#110

#1100

#1103

#1105

#1113

#1114

#1115

#1117

#1118

#1119

#112

#1121

#1122

#1125

#1128

#1129

#113

#1131

#1132

#1133

#1135

#1137

#1140

#1141

#1143

#1144

#1145

#1146

#1148

#1149

#115

#1152

#1153

#1154

#1155

#1156

#1157

#1158

#1159

#116

#1163

#1164

#1166

#1173

#1174

#1176

#1178

#1180

#1189

#1191

#1192

#1193

#1194

#1196

#1197

#1199

#12

#1200

#1202

#1204

#1205

#1206

#1208

#1209

#121

#1210

#1211

#1212

#1213

#1214

#1215

#1216

#1217

#1220

#1222

#1225

#1228

#1229

#123

#1230

#1230

#1231

#1231

#1233

#1234

#1235

#1240

#1241

#1242

#1246

#1249

#1250

#1251

#1253

#1257

#1259

#1260

#1263

#1265

#1267

#1270

#1271

#1272

#1273

#1276

#1277

#1278

#1279

#1280

#1283

#1287

#129

#1291

#1294

#1295

#1297

#1298

#1299

#1300

#1301

#1302

#1304

#1305

#1308

#1311

#1312

#1314

#1315

#1316

#1319

#1321

#1322

#1323

#1324

#1325

#1326

#133

#1330

#1331

#1332

#1336

#1338

#1339

#134

#1340

#1342

#1346

#1348

#1349

#1350

#1353

#1358

#1359

#136

#1364

#1365

#1367

#1370

#1371

#1375

#1375

#1377

#1377

#1383

#1383

#1385

#1386

#1387

#1388

#1389

#1390

#14

#140

#141

#144

#145

#147

#149

#151

#152

#153

#154

#157

#158

#159

#160

#161

#162

#163

#164

#165

#166

#167

#169

#171

#172

#173

#174

#176

#177

#178

#180

#183

#184

#186

#187

#189

#19

#190

#191

#192

#193

#195

#197

#198

#2

#200

#201

#202

#203

#205

#208

#211

#213

#214

#215

#219

#22

#221

#222

#227

#229

#23

#231

#234

#235

#237

#238

#24

#240

#241

#242

#243

#245

#248

#250

#251

#252

#253

#254

#255

#257

#260

#263

#264

#265

#266

#269

#27

#270

#271

#272

#274

#275

#276

#278

#282

#284

#285

#287

#290

#291

#292

#293

#294

#295

#30

#300

#301

#302

#303

#306

#307

#308

#309

#310

#311

#312

#315

#318

#319

#32

#320

#321

#325

#326

#33

#331

#333

#335

#337

#338

#340

#342

#343

#347

#350

#351

#352

#353

#354

#358

#36

#360

#361

#363

#364

#365

#366

#369

#37

#373

#375

#377

#378

#379

#380

#386

#387

#388

#389

#391

#392

#393

#396

#397

#398

#399

#40

#401

#405

#408

#409

#41

#41

#411

#413

#414

#415

#417

#418

#419

#42

#420

#421

#424

#426

#427

#429

#43

#431

#432

#433

#439

#441

#443

#445

#446

#449

#45

#450

#451

#453

#455

#457

#458

#461

#462

#466

#467

#468

#469

#47

#470

#471

#472

#474

#475

#479

#48

#482

#483

#483

#486

#489

#491

#494

#495

#496

#497

#498

#5

#50

#501

#502

#503

#505

#506

#509

#51

#510

#514

#515

#516

#518

#519

#52

#520

#521

#522

#523

#528

#53

#530

#531

#534

#536

#539

#541

#544

#545

#546

#547

#548

#549

#55

#552

#555

#558

#562

#563

#565

#566

#569

#570

#571

#572

#573

#574

#577

#578

#580

#581

#585

#589

#590

#591

#592

#595

#596

#599

#602

#603

#604

#608

#609

#610

#611

#613

#614

#62

#621

#623

#628

#632

#633

#634

#636

#639

#64

#640

#643

#644

#645

#648

#650

#651

#654

#657

#66

#661

#665

#666

#667

#668

#670

#673

#674

#675

#676

#679

#680

#681

#682

#683

#684

#685

#687

#688

#69

#690

#691

#693

#694

#697

#698

#7

#701

#702

#703

#705

#707

#708

#711

#712

#715

#716

#717

#719

#720

#721

#729

#73

#731

#735

#736

#739

#74

#740

#743

#744

#746

#749

#75

#752

#753

#758

#76

#760

#763

#766

#77

#770

#773

#775

#778

#779

#78

#782

#789

#79

#790

#792

#793

#794

#797

#798

#8

#80

#800

#802

#803

#806

#807

#810

#813

#817

#818

#82

#821

#822

#824

#825

#827

#828

#830

#831

#833

#835

#837

#838

#839

#84

#840

#85

#851

#852

#853

#855

#856

#857

#86

#863

#867

#87

#871

#877

#879

#88

#882

#885

#886

#888

#889

#89

#890

#891

#895

#898

#899

#90

#902

#903

#904

#905

#906

#907

#911

#913

#914

#915

#915

#920

#923

#926

#93

#931

#932

#935

#936

#937

#94

#940

#942

#945

#946

#948

#949

#954

#955

#956

#957

#96

#960

#961

#962

#963

#965

#969

#97

#971

#971

#973

#979

#98

#981

#983

#984

#989

#99

#990

#991

#993

#995

#996

#998

2e6a6c32ae Update README.md Ashraful Islam 2023-12-12 18:26:13 -06:00
2206e8f7d9 Update convert.py Merrick Christensen 2023-12-12 14:33:33 -07:00
e0a53edb46 llama v1 request Awni Hannun 2023-12-12 13:32:05 -08:00
f0c57c1361 llama v2 with sharded weights Awni Hannun 2023-12-12 12:48:15 -08:00
9a02dce35c Merge pull request #80 from 805karansaini/main Awni Hannun 2023-12-12 12:20:13 -08:00
eae9431143 Typo Fix 805karansaini 2023-12-13 01:45:50 +05:30
9b3f64d196 Merge pull request #75 from ml-explore/mixtral Awni Hannun 2023-12-12 10:41:25 -08:00
2439333a57 fixed doc for ResNet Sarthak Yadav 2023-12-12 19:07:39 +01:00
f37e777243 added CIFAR10 + ResNet example Sarthak Yadav 2023-12-12 19:01:06 +01:00
034d0cfc2e nit Awni Hannun 2023-12-12 08:42:32 -08:00
0f66a12721 typos in readme Awni Hannun 2023-12-12 08:41:28 -08:00
2ffd0da009 mixtral runs a bit faster Awni Hannun 2023-12-12 08:36:40 -08:00
94705ed38b Add large v3 bofenghuang 2023-12-12 17:26:52 +01:00
e42682dced initial mixtral Awni Hannun 2023-12-12 07:44:23 -08:00
6e723a015a whisper default in fp16 Awni Hannun 2023-12-12 07:37:35 -08:00
13f1142eaa Merge pull request #73 from jj701/mnist-requirements-txt Awni Hannun 2023-12-11 19:37:53 -08:00
bd742ec03c Adding Requirements.txt jj701 2023-12-11 20:45:39 -06:00
5be26ae91b Merge pull request #69 from TristanBilot/main Awni Hannun 2023-12-11 14:22:47 -08:00
b606bfa6a7 fix comments before merge Tristan Bilot 2023-12-11 23:10:46 +01:00
b95e48e146 use tree_flatten within L2 regularization Tristan Bilot 2023-12-11 20:15:11 +01:00
ed5a830626 add GCN implementation Tristan Bilot 2023-12-11 17:48:07 +01:00
ecd96acfe4 Merge pull request #66 from Haixing-Hu/fix-issue-54 Awni Hannun 2023-12-10 18:57:51 -08:00
5b62270556 fix: fix issue #54, use CPU device to load the Torch model Haixing Hu 2023-12-11 10:54:55 +08:00
a4d932bf26 fix conversion Awni Hannun 2023-12-10 14:35:39 -08:00
2652b4f055 Merge pull request #52 from ricardo-larosa/mistral_batch_size Awni Hannun 2023-12-10 11:25:23 -08:00
ce9ba916a3 Add arg tokens_per_eval for token generation ricardo-larosa 2023-12-09 19:43:44 +01:00
a577abc313 Cleaner masking code Joe Barrow 2023-12-09 21:21:24 -05:00
3a3ea3cfb0 Merge pull request #53 from ml-explore/mistral_lora Awni Hannun 2023-12-09 15:05:29 -08:00
5a5decf767 revert accidental change Awni Hannun 2023-12-09 14:58:45 -08:00
036090f508 few more nits Awni Hannun 2023-12-09 14:20:19 -08:00
98f4346c81 black format Awni Hannun 2023-12-09 14:15:25 -08:00
b8332a1e66 generalize lora finetuning for llama and mistral Awni Hannun 2023-12-09 14:13:55 -08:00
5d4838b02e Updating BERT model to take advantage of bias param in MultiHeadAttention Joe Barrow 2023-12-09 12:07:33 -05:00
46c6bbe0a1 Merge pull request #43 from jbarrow/main Awni Hannun 2023-12-09 09:03:49 -08:00
d873e10dfe Updating README for current example, making python>=3.8 compatibile, and fixing code type Joe Barrow 2023-12-09 12:01:58 -05:00
20d920a7eb Requirements for running BERT Joe Barrow 2023-12-09 10:52:55 -05:00
45ca4ed3f6 Updating README Joe Barrow 2023-12-09 10:48:34 -05:00
7320456226 Cleaning implementation for merge Joe Barrow 2023-12-09 10:41:15 -05:00
8b2a6fee33 Merge pull request #50 from eltociear/patch-1 Awni Hannun 2023-12-08 17:00:25 -08:00
ba0bd65510 Update CONTRIBUTING.md Ikko Eltociear Ashimine 2023-12-09 08:02:34 +09:00
331690491f Merge pull request #45 from bbelescot/clarify-llama-readme-instructions Awni Hannun 2023-12-08 09:38:07 -08:00
5bdd030387 📝 apply the path change to the convert cmd for consistency bbelescot 2023-12-08 17:11:50 +01:00
e05ee57bab Update README for mlx-examples repo Joe Barrow 2023-12-08 10:20:50 -05:00
6259c9a048 Merge pull request #40 from Jacksonzhang0316/main Awni Hannun 2023-12-08 06:48:21 -08:00
bbc18a3240 Merge pull request #47 from nepalivai/patch-1 Awni Hannun 2023-12-08 06:47:12 -08:00
743f0a2729 Update README.md Nepalivai 2023-12-08 19:41:38 +05:45
68c4282766 📝 clarify python command for llama example bbelescot 2023-12-08 13:55:26 +01:00
53f7e68d7e fix typo in README.md biondi 2023-12-08 18:07:16 +08:00
4e5b8ceafe BERT implementation Joe Barrow 2023-12-08 05:14:11 -05:00
4018aed335 fix: Unsupported BFloat16 Data Type Issue with MPS Backend 张嘉豪 2023-12-08 16:19:35 +08:00
ff3cc56c8d Merge pull request #37 from robertmccraith/main Awni Hannun 2023-12-07 13:35:31 -08:00
4ed942d7f5 fix: typo in variable name Robert McCraith 2023-12-07 21:30:04 +00:00
172a60056f update whisper readme and requirements Awni Hannun 2023-12-07 11:15:54 -08:00
54952a0d80 Merge pull request #12 from chatgpt-1/main Awni Hannun 2023-12-07 08:53:30 -08:00
a2b837e0f5 Merge pull request #30 from HonorRong/patch-1 Awni Hannun 2023-12-07 08:13:37 -08:00
e5cb9635a7 Merge pull request #32 from LeTarrask/main Awni Hannun 2023-12-07 08:12:26 -08:00
12ef61915a Merge pull request #33 from tensojka/patch-1 Awni Hannun 2023-12-07 08:07:11 -08:00
62dcb3301f Merge pull request #10 from ricardo-larosa/fix-unsupported-scalartype Awni Hannun 2023-12-07 08:05:44 -08:00
85345d42cb Merge branch 'ml-explore:main' into fix-unsupported-scalartype Ricardo La Rosa 2023-12-07 17:04:01 +01:00
71aff8c346 Fix unsupported ScalarType BFloat16 ricardo-larosa 2023-12-06 13:30:59 +01:00
5a0c36d8b6 Fix link in docstring in wikitext function Ondřej Sojka 2023-12-07 14:19:55 +01:00
9cddf4f2cd fixed typo Alex Luna 2023-12-07 10:19:37 +00:00
ec97c7531b Update README.md waterstone 2023-12-07 16:44:29 +08:00
1289f0bd9c Merge pull request #22 from Vaibhavs10/add_hf_url Awni Hannun 2023-12-06 21:15:42 -08:00
76c78aa486 Merge pull request #23 from adhishthite/main Awni Hannun 2023-12-06 21:14:41 -08:00
443037859a Merge pull request #5 from jaglinux/patch-1 Awni Hannun 2023-12-06 21:07:13 -08:00
e55c18d164 fix(sd): Add mlx as a dependency David Chavez 2023-12-06 12:08:43 +01:00
a1b4a475e1 Clarification in the stable diffusion readme Angelos Katharopoulos 2023-12-06 15:22:46 -08:00
e488831e03 bug fix in transformer_lm example Leon Ericsson 2023-12-06 17:42:23 +01:00
8b965b2e33 Add missing numpy dependency Hitarth Sharma 2023-12-07 08:21:21 +13:00
9cf82a0d43 Benchmark all models if user allows. adhishthite 2023-12-07 00:07:42 +05:30
5a5b2b6dcf Add url to mlx checkpoints on HF. Vaibhav Srivastav 2023-12-06 19:13:14 +01:00
6cbc029450 Fix timestamp extraction bug in transcribe function crackerben 2023-12-06 20:34:30 +08:00
429ddb30dc Fix unsupported ScalarType BFloat16 ricardo-larosa 2023-12-06 13:30:59 +01:00
5fc117025a Update README.md Jagadish Krishnamoorthy 2023-12-06 00:30:37 -08:00
0bf5d0e3bc Merge pull request #2 from ml-explore/mistral Awni Hannun 2023-12-05 21:44:35 -08:00
bcede4bc0c comments Awni Hannun 2023-12-05 21:36:47 -08:00
e4333a3325 update readme Awni Hannun 2023-12-05 11:58:58 -08:00
234a5f5cfe nits Awni Hannun 2023-12-05 11:24:30 -08:00
b7840a4721 mistral Awni Hannun 2023-12-05 11:02:52 -08:00
1900564f59 format Awni Hannun 2023-11-30 11:52:47 -08:00
5117e2e65d Merge pull request #1 from ml-explore/copyright Awni Hannun 2023-11-30 11:16:44 -08:00
31bc57c4ff add copyright in source Awni Hannun 2023-11-30 11:08:53 -08:00
5d6353aab7 lora Awni Hannun 2023-11-29 14:14:11 -08:00
4950e9f374 contribution + code of conduct Awni Hannun 2023-11-29 12:31:18 -08:00
b364cc56cd Add the Llama and Stable Diffusion examples Angelos Katharopoulos 2023-11-29 10:38:20 -08:00
7f1328f333 fix readme Awni Hannun 2023-11-29 08:19:30 -08:00
b243c1d8f4 a few examples Awni Hannun 2023-11-29 08:17:26 -08:00
e31d82d3ed Initial commit Awni Hannun 2023-11-28 15:37:50 -08:00