Commit Graph

Select branches

Hide Pull Requests

0.3

0.3.1

0.4

2d_example

71844_migrate_master_branch_to_main

brianjo-patch-1

brianjo-patch-2

brianjo-patch-3

chirag/fix-fsdp-tutorial

chirag/fix-python-tests

ci_test

circleci-project-setup

classroomtestfix

ddp-tutorial-code

ddp-tutorial-setdevice

example_script

fix-ci-rl-error

fix_ci

fix_example

gh-pages

gh/bowangbj/1/base

gh/bowangbj/1/orig

gh/weifengpy/1/base

gh/weifengpy/1/head

gh/weifengpy/1/orig

gh/weifengpy/2/base

gh/weifengpy/2/head

gh/weifengpy/2/orig

main

mnist

msaroufim-patch-1

msaroufim-patch-2

no_pg_destroy_warning

owner

pbelevich-patch-1

perftests

publish-examples

publish-examples2

regression

reinforce

revert-1003-feat/add_siamese_network_example

revert-1018-dev/yuliya/mps_device

revert-1346-main

revert-1359-add-mass-spring-example

revert-386-RL-loss-normalized

revert-529-refactor_print_progress

revert-81-flush_print

rnn

sdym/log1p

sdym/pretrained

subramen-patch-1

subramen-patch-2

suraj813-patch-1

suraj813-patch-2

svekars-patch-1

tp_example

update-publishing-yml

update_batch_rpc

update_rpc

v0.2.0

vae_fix

#1

#100

#1000

#1001

#1001

#1003

#1005

#1006

#1007

#1008

#1009

#101

#101

#1011

#1014

#1016

#1017

#1018

#1019

#102

#102

#1021

#1024

#1025

#1026

#1027

#1028

#1031

#1032

#1034

#104

#1042

#1043

#1044

#1045

#1048

#1051

#1057

#1058

#1061

#1062

#1064

#1065

#1066

#1067

#1068

#1069

#1070

#1072

#1073

#1076

#1079

#108

#1083

#1084

#1086

#109

#1091

#1092

#1094

#1095

#1098

#1099

#11

#1100

#1102

#1103

#1104

#1106

#1107

#1108

#1109

#111

#1110

#1113

#1114

#1115

#1117

#1118

#1118

#1119

#1120

#1124

#1125

#1126

#1127

#1127

#1129

#1130

#1130

#1135

#1137

#1138

#1139

#114

#1140

#1141

#1142

#1145

#1149

#1152

#1155

#1156

#1157

#1157

#1158

#1158

#1159

#1160

#1161

#1162

#1162

#1163

#1164

#1169

#117

#1172

#1173

#1174

#1176

#118

#1181

#1187

#1189

#119

#119

#1191

#1191

#1192

#1192

#1193

#1197

#1198

#1201

#1203

#1204

#1205

#1205

#1206

#1207

#1209

#121

#1212

#1212

#1214

#1215

#1216

#1217

#1219

#122

#1220

#1221

#1222

#1223

#1224

#1225

#1226

#1230

#1231

#1232

#1234

#1237

#1237

#1239

#1239

#124

#1240

#1243

#1244

#1246

#1246

#1247

#1248

#125

#1250

#1251

#1252

#1254

#1254

#1256

#1258

#1259

#1260

#1261

#1262

#1262

#1265

#1265

#1266

#1267

#1268

#1269

#127

#1272

#1273

#1274

#1275

#1276

#1276

#128

#1280

#1281

#1282

#1283

#1283

#1284

#1285

#1286

#1287

#1288

#129

#1290

#1290

#1291

#1291

#1293

#1293

#1294

#1296

#1297

#1299

#1300

#1300

#1301

#1301

#1303

#1303

#1304

#1305

#1305

#1306

#1309

#131

#131

#1311

#1312

#1314

#1315

#1316

#1317

#1318

#132

#1320

#1321

#1322

#1323

#1324

#1326

#1327

#1328

#133

#1330

#1332

#1333

#1334

#1335

#1336

#1337

#1338

#1339

#1340

#1341

#1341

#1342

#1343

#1344

#1345

#1346

#1347

#1348

#1349

#135

#1350

#1354

#1356

#1357

#1358

#1359

#136

#136

#1360

#1361

#1363

#1364

#1365

#1366

#1367

#1368

#1369

#1370

#1371

#1372

#1373

#1374

#1375

#1376

#1379

#1380

#1381

#1382

#1383

#1384

#1385

#1386

#1387

#1388

#1388

#1389

#1389

#139

#139

#1390

#1390

#1391

#1391

#1393

#1393

#1394

#1396

#1396

#1398

#1399

#140

#1400

#1401

#1402

#1406

#1406

#1407

#1407

#1408

#1408

#1409

#141

#1410

#1411

#1411

#1412

#1412

#1413

#1414

#1415

#1415

#1416

#1416

#1417

#1417

#1418

#1418

#145

#149

#149

#15

#150

#153

#154

#16

#160

#17

#172

#172

#175

#176

#177

#179

#18

#182

#183

#184

#184

#185

#186

#188

#189

#19

#190

#191

#191

#192

#193

#196

#197

#197

#2

#2

#20

#200

#202

#203

#208

#209

#21

#21

#212

#212

#213

#216

#218

#219

#22

#22

#220

#222

#223

#223

#226

#227

#228

#229

#23

#230

#233

#235

#235

#238

#239

#24

#242

#244

#245

#249

#25

#250

#251

#259

#26

#260

#262

#265

#266

#268

#269

#27

#270

#272

#272

#273

#273

#279

#28

#283

#284

#287

#287

#288

#29

#293

#296

#299

#3

#30

#300

#301

#302

#306

#306

#307

#309

#31

#310

#313

#313

#318

#319

#32

#320

#321

#322

#324

#325

#325

#328

#33

#331

#332

#333

#333

#335

#336

#337

#338

#339

#339

#34

#340

#340

#344

#345

#346

#348

#351

#353

#354

#36

#36

#360

#360

#363

#364

#364

#365

#366

#37

#37

#371

#371

#375

#379

#38

#381

#386

#39

#390

#392

#394

#397

#398

#4

#40

#401

#402

#402

#403

#404

#404

#407

#408

#41

#413

#414

#416

#416

#417

#419

#42

#420

#422

#422

#423

#424

#426

#427

#432

#436

#437

#44

#441

#442

#444

#446

#446

#449

#45

#45

#451

#454

#455

#46

#462

#463

#464

#466

#468

#469

#47

#474

#477

#48

#484

#485

#489

#490

#492

#493

#494

#495

#496

#5

#50

#506

#506

#508

#509

#511

#513

#515

#519

#52

#520

#523

#525

#528

#529

#530

#535

#54

#541

#545

#546

#551

#551

#555

#556

#560

#562

#565

#569

#57

#572

#575

#577

#58

#583

#585

#587

#589

#59

#591

#596

#60

#60

#601

#602

#61

#610

#612

#614

#616

#62

#625

#626

#627

#629

#632

#632

#633

#634

#637

#644

#647

#65

#654

#658

#66

#662

#664

#667

#669

#67

#670

#673

#677

#678

#68

#683

#684

#686

#687

#688

#69

#690

#692

#692

#693

#694

#697

#699

#7

#701

#702

#702

#703

#704

#705

#707

#708

#709

#710

#711

#712

#713

#719

#72

#72

#721

#722

#722

#724

#728

#728

#73

#731

#732

#733

#735

#74

#743

#745

#746

#748

#749

#750

#752

#758

#76

#762

#763

#764

#765

#767

#773

#774

#775

#776

#777

#777

#779

#78

#78

#781

#785

#786

#787

#789

#79

#791

#792

#794

#796

#798

#8

#80

#800

#801

#803

#804

#807

#808

#809

#81

#810

#811

#815

#82

#820

#823

#825

#825

#828

#832

#833

#839

#84

#841

#841

#842

#844

#845

#849

#85

#857

#859

#860

#861

#863

#864

#867

#868

#868

#871

#874

#875

#876

#877

#879

#879

#881

#882

#883

#884

#886

#888

#890

#890

#891

#893

#894

#896

#897

#899

#90

#90

#904

#906

#907

#908

#909

#91

#912

#916

#917

#919

#92

#923

#924

#93

#93

#931

#935

#939

#94

#944

#945

#947

#948

#949

#953

#96

#968

#97

#97

#970

#971

#972

#973

#974

#975

#977

#978

#979

#98

#98

#980

#981

#982

#985

#986

#988

#99

#99

#990

#992

#993

#994

#995

#996

#997

#998

#999

Mono Color

• acc295dc7b Clean up README.md by removing badge and whitespace main Mark Saroufim 2025-09-01 10:21:46 -07:00
• 77f55b9121 silence destroy_proces_group() warning (#1387) Mark Saroufim 2025-08-25 13:18:39 -07:00
• 6079ce615e silence destroy_proces_group() warning no_pg_destroy_warning Mark Saroufim 2025-08-25 13:16:59 -07:00
• 746c0a2d7e Modernize distributed/rpc/pipeline (#1385) Mark Saroufim 2025-08-25 13:01:44 -07:00
• 7fce8bb0da Unbreak torch/distributed/rpc/batch (#1386) Mark Saroufim 2025-08-25 13:01:33 -07:00
• 3b1a2d16c6 Merge branch 'main' into update_batch_rpc update_batch_rpc Mark Saroufim 2025-08-25 12:55:52 -07:00
• 47980ab8ee update Mark Saroufim 2025-08-25 12:55:03 -07:00
• e5f34d1ca8 More RPC examples Mark Saroufim 2025-08-25 12:54:14 -07:00
• 1a24d26a3e Fix More RPC examples Mark Saroufim 2025-08-25 12:53:50 -07:00
• f9799096c2 update update_rpc Mark Saroufim 2025-08-25 12:24:51 -07:00
• fce10a508c update Mark Saroufim 2025-08-25 12:22:28 -07:00
• 89e1215636 Modernize RPC example Mark Saroufim 2025-08-25 12:10:45 -07:00
• f1723eb8f3 Clean up commented paths in run_python_examples.sh (#1384) Mark Saroufim 2025-08-23 10:32:38 -07:00
• ef95195574 Clean up commented paths in run_python_examples.sh msaroufim-patch-2 Mark Saroufim 2025-08-23 09:18:43 -07:00
• 7de8d218df Comment out super resolution paths and run command Mark Saroufim 2025-08-23 00:03:28 -07:00
• 4a3b6e34f6 Revert "Add Swin Transformer Example " (#1383) Mark Saroufim 2025-08-22 23:21:04 -07:00
• 302283698e Revert "Add Swin Transformer Example (#1346)" revert-1346-main Mark Saroufim 2025-08-22 23:20:55 -07:00
• 0e5d2ed8c2 Deploying to gh-pages from @ 89c7d7588d 🚀 gh-pages msaroufim 2025-08-23 06:12:12 +00:00
• 89c7d7588d Upgrade to Gymnasium rather than Gym (#1381) Mark Towers 2025-08-23 07:11:42 +01:00
• 32fdb4952f Fix argument help descriptions to match default values (#1281) Nariaki Tateiwa 2025-08-23 15:11:21 +09:00
• 28d16ffaa5 Add AdamW optimizer support for World Language Model example (#1380) Sang Park 2025-08-23 15:08:25 +09:00
• 993a98a67a Fix invalid characters in filename in neural_style.py (line 119 in train function) (#1294) Innoy 2025-08-23 14:06:24 +08:00
• 892eca0a94 Add Swin Transformer Example (#1346) Sumantro Mukherjee 2025-08-23 11:30:14 +05:30
• a630ec6965 Fix license in environment.py Mark Saroufim 2025-08-13 11:32:39 -07:00
• 28e586404f Add accelerate API support for Word Language Model example (#1345) Francisco R Castro Garcia 2025-08-13 12:31:07 -06:00
• 5a4ca9269f Adding torch accelerator and requirements file to FSDP2 example (#1375) Dianana 2025-08-05 19:37:50 -06:00
• e9a4e7510c Add accelerator API to RPC distributed examples: ddp_rpc, parameter_server, rnn (#1371) Jaime Fraustro 2025-07-27 13:15:30 -06:00
• 99f5c4e224 mark fsdp1 as deprecated and point to fsdp2 (#1374) Wei (Will) Feng 2025-07-20 23:48:25 -07:00
• 890a2f0af3 Deploying to gh-pages from @ 8c1555d83c 🚀 soumith 2025-07-16 15:34:22 +00:00
• 2174f26a9c fsdp_tp_example fsdp1-> fsdp2 Jiani Wang 2025-07-15 11:22:25 -07:00
• 8c1555d83c remove manual n_heads change Jiani Wang 2025-07-15 13:17:11 -07:00
• 58370dca54 Bump torch from 1.9.0 to 2.2.0 in /distributed/rpc/batch dependabot[bot] 2024-07-25 11:22:06 +00:00
• bd1254b79e Fixed "--help" output in README.md Anthony Smith 2024-12-31 13:08:49 -06:00
• 86ed9e411c Enhance README and examples for Tensor Parallelism - Added installation instructions and example running commands to README.md. - Update files to have a better organization jafraustro 2025-07-10 09:51:46 -07:00
• 698a89e5fc Add torch.accelerator API to mingGPT example (#34) gera-aldama 2025-07-11 13:18:00 -06:00
• f84bcb398d add min_gpu verification similart tu tensor_parallel_example.py example jafraustro 2025-07-11 08:17:29 -07:00
• eab402d04d Update requirements.txt to specify torch version >=2.7 jafraustro 2025-07-10 09:01:04 -07:00
• 330201265c Refactor DDP example to use Accelerator API jafraustro 2025-07-08 07:54:43 -07:00
• d86d205660 Switch to torchrun for distributed launches jafraustro 2025-07-08 07:53:39 -07:00
• d47f0f34f1 Add accelerator API to GCN example. jafraustro 2025-07-08 08:12:11 -07:00
• c04a5a10ba Update .github/workflows/main_distributed.yaml _githubsgi 2025-07-09 10:20:44 -07:00
• d3849110ac Update .github/workflows/main_distributed.yaml _githubsgi 2025-07-09 10:20:24 -07:00
• 26409468df Update .github/workflows/main_distributed.yaml _githubsgi 2025-07-09 10:20:08 -07:00
• 997c123c0e Updating Python to 3.10, fsdp_tp_example.py to accelerator Goswami, Subrata 2025-07-07 16:49:53 -07:00
• 3fc7853882 Updating comments in multiple files. Goswami, Subrata 2025-07-01 12:10:19 -07:00
• bac5d0f5e8 Removing pre's from requirement.txt, ... Goswami, Subrata 2025-07-01 11:17:21 -07:00
• a5b6297a21 Bumping up Python from 3.8 to 3.9 in wf yaml. sg 2025-07-01 03:16:29 +00:00
• fe39e70a63 Bumping up Pyhon version to 3.9 . sg 2025-06-30 19:08:29 +00:00
• 56cc1a6cc3 Updating CI torch version. sg 2025-06-27 01:58:20 +00:00
• 3055e3ea56 Moving CommDebugMode code to a seperate PR. sg 2025-06-25 00:55:51 +00:00
• f8c79f153a cuda to accelerator subrata goswami 2025-06-11 23:31:19 +00:00
• 6bfe651404 Cuda to accelerator, +CommDebugMode sg 2025-06-11 22:24:55 +00:00
• d0b7e37094 Update main.py tutorial, set persistent_workers, make GPU execution much faster Max Buckley 2025-07-09 21:50:20 +02:00
• 42068585f7 Fixed use_accel not defined issue (#1363) BL 2025-07-07 13:31:18 -05:00
• 16554e58b3 imagenet: fix typo addressing args.gpu (#1361) Dmitry Rogozhkin 2025-06-30 11:52:25 -07:00
• 8d408d2a50 Revert "Add Differentiable Physics: Mass-Spring System example" (#1360) Mark Saroufim 2025-06-25 13:20:01 -07:00
• ac6b73ef9c Revert "Add Differentiable Physics: Mass-Spring System example (#1359)" revert-1359-add-mass-spring-example Mark Saroufim 2025-06-25 13:19:49 -07:00
• 7c35995e4c Add Differentiable Physics: Mass-Spring System example (#1359) Abhishek Nandy 2025-06-25 21:16:09 +05:30
• 6f616144df Add accelerate API support for Super Resolution example (#1358) Francisco R Castro Garcia 2025-06-23 19:17:56 -06:00
• 2944a9dcc5 Change run_python_examples.sh to run word_language_model/generate.py after each call to word_language_model/main.py Francisco R Castro G 2025-05-20 00:04:45 +00:00
• 6595d7b461 Update word_language_model/generate.py to remove duplicates, use abc order Francisco R Castro Garcia 2025-05-19 17:55:34 -06:00
• a86bda00a1 Fix torch>= 2.6 generate.py compatibility in Word_language_example Francisco R Castro G 2025-05-19 22:14:49 +00:00
• 37986af69a Removing unneeded print Edgar Romo Montiel 2025-06-13 17:53:32 -07:00
• dcc2474838 Add Accelerator Api to Imagenet Example eromomon 2025-05-19 16:27:21 -07:00
• 3ddcc89e48 fix typo which make tgt tensor data wrong (#1356) Wenzhe Zhou 2025-06-16 14:05:24 +08:00
• fcce71c1f0 Restore default CI configuration for VAE and Siamese examples using accelerator API (#1342) eromomon 2025-06-15 17:00:35 -06:00
• 5cc81aae65 save model in global rank 0 in multinode (#1357) IceCreamMilkyTea 2025-06-16 06:58:17 +08:00
• abfa4f9cc4 Fix super_resolution example for torch>=2.6 (#1350) Francisco R Castro Garcia 2025-06-14 13:40:15 -06:00
• ac7e960b35 Support torch>=2.6 in word_language_model example (#1347) Dmitry Rogozhkin 2025-05-17 20:20:56 -07:00
• cc8e404cbc Use torch.accelerator in DCGAN example (#1344) Dianana 2025-05-13 13:14:47 -06:00
• f2674984ab Deploying to gh-pages from @ 7028a2ecc9 🚀 msaroufim 2025-05-13 19:09:00 +00:00
• 7028a2ecc9 This PR Improve docs build ci (#1336) oepeng 2025-05-13 21:08:26 +02:00
• 70922969e7 FSDP2 example code for tutorial (#1343) Wei (Will) Feng 2025-05-09 10:12:08 -07:00
• fbdea94a37 Update on "FSDP2 example" gh/weifengpy/1/head Wei Feng 2025-05-08 16:36:02 -07:00
• 9789d443e3 FSDP2 example gh/weifengpy/1/orig Wei Feng 2025-05-08 16:36:02 -07:00
• 4c78ad122c Update base for Update on "FSDP2 example" gh/weifengpy/1/base Wei Feng 2025-05-08 16:36:02 -07:00
• f55919e5e2 Update on "FSDP2 example" Wei Feng 2025-05-07 21:03:00 -07:00
• 7bd6449ffe Update base for Update on "FSDP2 example" Wei Feng 2025-05-07 21:03:00 -07:00
• 7d43f13e5e Update on "FSDP2 example" Wei Feng 2025-05-07 20:59:53 -07:00
• fe27fef5b4 Update base for Update on "FSDP2 example" Wei Feng 2025-05-07 20:59:53 -07:00
• 098ac2c122 Update on "FSDP2 example" Wei Feng 2025-05-07 19:49:11 -07:00
• ff5706f9c5 Update base for Update on "FSDP2 example" Wei Feng 2025-05-07 19:49:11 -07:00
• 4d2b62e6ad Update on "FSDP2 example" Wei Feng 2025-05-05 19:53:31 -07:00
• 259172f190 Update base for Update on "FSDP2 example" Wei Feng 2025-05-05 19:53:31 -07:00
• 48eb64b99a args gh/weifengpy/2/orig Wei Feng 2025-05-05 19:51:11 -07:00
• d14ce566f8 args gh/weifengpy/2/head Wei Feng 2025-05-05 19:51:09 -07:00
• 9708f36d66 FSDP2 example Wei Feng 2025-05-05 19:45:18 -07:00
• e5d4d63235 FSDP2 example gh/weifengpy/2/base Wei Feng 2025-05-05 19:45:16 -07:00
• 54e132ece5 Use torch.accelerator API in VAE example (#1338) eromomon 2025-05-02 10:40:02 -06:00
• 12dc18e872 Use torch.accelerator API in Siamese Network example (#1337) eromomon 2025-05-01 23:39:26 -06:00
• b7aebb5ad2 Update tensor_parallel_example.py (#1324) nrothGIT 2025-04-30 20:03:08 -04:00
• 2c435c7868 Use torch.accelerator API in GAT example (#1335) Jaime Fraustro 2025-04-30 16:28:55 -06:00
• 65722fe3ce Use torch.accelerator API in mnist examples (#1334) Jaime Fraustro 2025-04-30 08:02:42 -06:00
• d00125af85 Deploying to gh-pages from @ 6967ff5676 🚀 msaroufim 2025-04-29 18:48:30 +00:00
• 6967ff5676 Update doc-build.yml and requirements for documentation build (#1333) Jaime Fraustro 2025-04-29 12:47:49 -06:00
• 65afde6aab Use torch.accelerator API in Fast Neural Style example (#1327) Dmitry Rogozhkin 2025-04-28 12:46:10 -07:00
• 00ef8a78e4 Respect each example requirements and use uv (#1330) Dmitry Rogozhkin 2025-04-25 21:28:00 -07:00
• 8393ceb2ba Add support for Intel GPU to Fast Neural Style example (#1318) eromomon 2025-04-18 11:30:03 -06:00
• 5dfeb46902 fix typo (#1272) Kosi Asuzu 2025-02-09 16:56:58 +00:00
• 7e907f781f Update doc-build.yml svekars-patch-1 Svetlana Karslioglu 2025-01-29 14:29:53 -08:00