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

#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

Mono Color

• a7e414687e update create_dataset Goekdeniz-Guelmez 2025-02-04 10:45:23 +0100
• 7b0141455e better create_dataset Goekdeniz-Guelmez 2025-02-04 10:43:00 +0100
• 9ece9aea02
  Merge branch 'ml-explore:main' into adding-ppo-training Gökdeniz Gülmez 2025-02-04 10:33:47 +0100
• bd1a42ec2f adding args into dataset handling Goekdeniz-Guelmez 2025-02-04 10:22:34 +0100
• 7173840283 first succesfull training run Goekdeniz-Guelmez 2025-02-04 09:18:45 +0100
• 21d0ab6e8a
  fix deepseek sharding (#1242) Awni Hannun 2025-02-03 16:59:50 -0800
• 3b3c20f508 fix deepseek sharding Awni Hannun 2025-02-03 16:54:58 -0800
• 0989c073b0
  Optimizations for mamba1 (#1213) Gökdeniz Gülmez 2025-02-03 22:36:08 +0100
• c43fc7ce59 nits + even faster Awni Hannun 2025-02-03 12:53:11 -0800
• ca32424043 updates Goekdeniz-Guelmez 2025-02-03 21:57:26 +0100
• 54e295ea80 fix name funcs Goekdeniz-Guelmez 2025-02-03 19:56:11 +0100
• 06f9c29c94 print func name Goekdeniz-Guelmez 2025-02-03 19:47:40 +0100
• 40bca770ae fixes Goekdeniz-Guelmez 2025-02-03 19:43:49 +0100
• 05d921b788 optims Goekdeniz-Guelmez 2025-02-03 19:37:05 +0100
• d9924d08d1
  Fix no validation in lora (#1241) Awni Hannun 2025-02-03 09:55:24 -0800
• 10da47af36 Fix no validation in lora Awni Hannun 2025-02-03 06:52:53 -0800
• 1d9e4802f0 first working prototype, will try training out at home Goekdeniz-Guelmez 2025-02-03 12:05:29 +0100
• 23d75cd7ad starting fist training test run Goekdeniz-Guelmez 2025-02-03 10:08:28 +0100
• 41ff5364d7 Merge branch 'adding-GRPO-training' of https://github.com/Goekdeniz-Guelmez/mlx-examples into adding-GRPO-training Goekdeniz-Guelmez 2025-02-03 09:19:00 +0100
• a3ed632422 dataset wrapper done Goekdeniz-Guelmez 2025-02-03 09:13:17 +0100
• 2c96da5155
  Merge branch 'ml-explore:main' into adding-orpo-training Gökdeniz Gülmez 2025-02-03 09:07:52 +0100
• 94debd5ba8
  Merge branch 'ml-explore:main' into adding-ppo-training Gökdeniz Gülmez 2025-02-03 09:07:37 +0100
• 734d6f4a69
  Merge branch 'ml-explore:main' into adding-GRPO-training Gökdeniz Gülmez 2025-02-03 09:07:20 +0100
• d034ca369e adding function for R1 Goekdeniz-Guelmez 2025-02-03 08:26:42 +0100
• 9c2ef38d4d
  only download local shard (#1240) Awni Hannun 2025-02-02 13:58:44 -0800
• 2ac502176d only download local shard Awni Hannun 2025-02-02 12:56:46 -0800
• fbb51f651a small fix Goekdeniz-Guelmez 2025-02-01 16:08:52 +0100
• a03d434bb9 clean up Goekdeniz-Guelmez 2025-01-31 21:37:15 +0100
• 541677aa7f cleaning up Goekdeniz-Guelmez 2025-01-31 21:36:24 +0100
• 5998272ec2 cleaning up some namings Goekdeniz-Guelmez 2025-01-31 21:27:59 +0100
• 243c9621d9 update lora.py Goekdeniz-Guelmez 2025-01-31 21:10:44 +0100
• aa7a11c753 updates Goekdeniz-Guelmez 2025-01-31 17:38:01 +0100
• b379359385 small fix Goekdeniz-Guelmez 2025-01-31 17:19:55 +0100
• 595125ad4e updates Goekdeniz-Guelmez 2025-01-31 17:19:05 +0100
• 3ad6405298
  Merge branch 'ml-explore:main' into adding-ppo-training Gökdeniz Gülmez 2025-01-31 17:04:10 +0100
• 9e39c544eb initial commit Goekdeniz-Guelmez 2025-01-31 17:03:40 +0100
• a57d553fc1 update Goekdeniz-Guelmez 2025-01-31 16:57:43 +0100
• 80bcf68956 grpo_trainer shoudl be done Goekdeniz-Guelmez 2025-01-31 16:54:18 +0100
• 6c58aa995c updates Goekdeniz-Guelmez 2025-01-31 16:27:31 +0100
• b31d9cbb65 removing is-reference-free argument Goekdeniz-Guelmez 2025-01-31 00:01:43 +0100
• 93370ff1c3 updates ans fixing the KL div lines Goekdeniz-Guelmez 2025-01-30 23:55:34 +0100
• 70360d9d0d
  Merge ec06c04f4f into e8afb59de4 Sindhu Satish 2025-01-29 07:42:19 -0800
• ec06c04f4f revert revision changes and retain qwen2 support Sindhu Satish 2025-01-29 07:37:58 -0800
• ba6c7d3aba Qwen2 support Sindhu Satish 2025-01-29 07:30:11 -0800
• b0520e7708 Bug fix - Qwen2 support Sindhu Satish 2025-01-29 06:21:36 -0800
• b1e573d6e8
  Merge branch 'ml-explore:main' into adding-GRPO-training Gökdeniz Gülmez 2025-01-29 15:07:52 +0100
• b3d6fc38cd
  Merge branch 'ml-explore:main' into adding-dpo-training Gökdeniz Gülmez 2025-01-29 15:07:37 +0100
• ceccb4c9e9
  Merge branch 'ml-explore:main' into adding-orpo-training Gökdeniz Gülmez 2025-01-29 15:07:22 +0100
• 57e10446b0
  Merge branch 'ml-explore:main' into adding-support-for-mamba2 Gökdeniz Gülmez 2025-01-29 15:07:11 +0100
• a7e35ee748
  Merge branch 'ml-explore:main' into optimizations-for-mamba1 Gökdeniz Gülmez 2025-01-29 15:07:00 +0100
• 651f9a5cf8
  Merge dd1690df81 into e8afb59de4 Sindhu Satish 2025-01-29 06:01:07 -0800
• dd1690df81 bug fix Sindhu Satish 2025-01-29 06:00:06 -0800
• e89a131668 Include revision version for HF models while loading Sindhu Satish 2025-01-29 05:53:18 -0800
• 5e0ae83487 initial commit, gn Goekdeniz-Guelmez 2025-01-29 00:19:07 +0100
• e8afb59de4
  better overflow correction (#1229) Awni Hannun 2025-01-28 14:37:30 -0800
• 001323070a better overflow correction Awni Hannun 2025-01-28 12:27:45 -0800
• 6a367fa31e update Copyright to this year Goekdeniz-Guelmez 2025-01-28 21:04:03 +0100
• 0d4f2c4dc0 cleaning up and adding apple copyright to helium modelfile Goekdeniz-Guelmez 2025-01-28 21:02:50 +0100
• a928bba375 Pass down TrainingArgs instance to iterate_batches function and TrainingCallback methods Addresses #1224 Chime Ogbuji 2025-01-28 14:50:38 -0500
• 7b29cf0eda
  Merge branch 'ml-explore:main' into optimizations-for-mamba1 Gökdeniz Gülmez 2025-01-28 20:32:33 +0100
• 7a83077cd7
  chore(mlx-lm): support text type content in messages (#1225) Anchen 2025-01-28 12:13:50 +1100
• d98bf6f798 nits + format Awni Hannun 2025-01-27 16:43:51 -0800
• f44a52e2dc
  batched min p and fix spec gen sampling (#1222) Awni Hannun 2025-01-27 15:40:31 -0800
• 07f3d7d6bb chore: optimize the messagef content processing anchen 2025-01-27 09:37:29 +1100
• 649d3f82ae fix ACKNOWLEDGMENTS Goekdeniz-Guelmez 2025-01-26 17:05:34 +0100
• 9e5482ee74
  Merge branch 'main' into adding-dpo-training Gökdeniz Gülmez 2025-01-26 17:01:37 +0100
• 3642a9df9b update ACKNOWLEDGMENTS Goekdeniz-Guelmez 2025-01-26 17:00:32 +0100
• 294d189eed
  Merge branch 'main' into adding-orpo-training Gökdeniz Gülmez 2025-01-26 16:59:37 +0100
• de856c7223
  Merge branch 'main' into adding-support-for-mamba2 Gökdeniz Gülmez 2025-01-26 16:58:06 +0100
• 77faa14ba4
  adding support for kyutai's helium (#1208) Gökdeniz Gülmez 2025-01-26 16:19:07 +0100
• 5acb03d7bf fixes / nits Awni Hannun 2025-01-26 07:06:21 -0800
• 557649d8da removing tokenizer and updates Goekdeniz-Guelmez 2025-01-26 15:25:27 +0100
• 2f2ddd4811 clean up Goekdeniz-Guelmez 2025-01-26 15:17:06 +0100
• 4d0e52f7c8 more metrics Goekdeniz-Guelmez 2025-01-26 15:09:55 +0100
• 74ae24b883 chore(mlx-lm): support text type content anchen 2025-01-27 00:00:10 +1100
• 0ff1289bd9 updates Goekdeniz-Guelmez 2025-01-25 22:03:32 +0100
• d8e7834345 Removed rejected_rewards handling, Updated batch unpacking to match iterator, Updated batch unpacking to match iterator, Added preference score scaling, Simplified reward calculation, Removed redundant rejected_rewards Goekdeniz-Guelmez 2025-01-25 21:35:37 +0100
• 86b315fdf9 nits and quality of life improvements Goekdeniz-Guelmez 2025-01-24 22:40:27 +0100
• 531c3345c6 nits Goekdeniz-Guelmez 2025-01-24 18:13:05 +0100
• 54fcd8ed63 update DPODataset and added in system field too Goekdeniz-Guelmez 2025-01-24 18:11:56 +0100
• 09ed837896 updates Goekdeniz-Guelmez 2025-01-24 16:57:18 +0100
• e3688293ed removing dpo and fixing some stuff for orpo Goekdeniz-Guelmez 2025-01-24 16:09:22 +0100
• 677b6a5176 batched min p and fix spec gen sampling Awni Hannun 2025-01-23 15:20:21 -0800
• f787c08585 comments dist-eval Alex Barron 2025-01-23 06:36:31 -0800
• d5f49d65b9 ordering Alex Barron 2024-12-19 00:08:28 -0800
• 4385363c0f distributed evaluate Alex Barron 2024-12-18 22:12:08 -0800
• a1aace4d99
  Add the possibility to cache model instead of loading from disk each time. Haixuan Xavier Tao 2025-01-23 11:55:12 +0100
• 2462a34194 removig sanitize Goekdeniz-Guelmez 2025-01-22 22:30:15 +0100
• 0bb001121e niits Goekdeniz-Guelmez 2025-01-22 21:39:29 +0100
• aefe4ba160 nits Goekdeniz-Guelmez 2025-01-22 21:36:56 +0100
• e1d549bcd3 nits Goekdeniz-Guelmez 2025-01-22 21:03:21 +0100
• b0ece88909 nits Goekdeniz-Guelmez 2025-01-22 20:54:20 +0100
• 44809b3ead
  Merge branch 'ml-explore:main' into optimizations-for-mamba1 Gökdeniz Gülmez 2025-01-22 14:19:22 +0100
• dd29e74b89
  Merge branch 'ml-explore:main' into adding-support-for-mamba2 Gökdeniz Gülmez 2025-01-22 14:19:06 +0100
• 520f4c468f
  Merge branch 'ml-explore:main' into adding-support-for-helium Gökdeniz Gülmez 2025-01-22 14:18:51 +0100
• 4098c3bd2f
  Merge branch 'ml-explore:main' into adding-orpo-training Gökdeniz Gülmez 2025-01-22 14:18:38 +0100
• 69a8f11f7b
  Merge branch 'ml-explore:main' into adding-dpo-training Gökdeniz Gülmez 2025-01-22 14:18:24 +0100
• 9a3ddc3e65
  some fixes for pipeline parallel deep seek r1 (#1216) Awni Hannun 2025-01-21 19:40:29 -0800
• a4b716e65d small optimization Goekdeniz-Guelmez 2025-01-22 00:15:02 +0100
• df1406735b
  Fix dataset variable name, in datasets.py (#1212) Victor Nogueira 2025-01-21 23:12:43 +0100