All pull requests

Pull requests

• Zero gradients between steps in the tensor parallelism examples

  cla signed
  #1424·HimanshuJanbandhu opened Aug 18, 2026·

  ·
  2

• Generalize device declaration in mnist_hogwild and reinforcement_learning

  cla signed
  #1423·pgmoka opened Aug 6, 2026·

  ·
  10

• distributed/rpc/rl: compute REINFORCE returns per observer

  cla signed
  #1422·getchebarne opened May 12, 2026·

  ·
  1

• Update DDP tutorial PyTorch requirement

  cla signed
  #1421·Shy4n7 opened May 6, 2026·

  ·
  1

• Preserve explicit ImageNet resume learning rate

  cla signed
  #1420·Shy4n7 opened May 6, 2026·

  ·
  4

• Refactor MNIST forward style and improve device selection

  cla signed
  #1419·amit-chaubey opened Apr 13, 2026·

  ·
  1

• vae example: provide an additional --xla flag to be able to run this test on TPU if available

  cla signed
  #1416·fedebongio opened Mar 30, 2026·

  ·
  1

• Provide an additional --xla flag to be able to run this test on TPU if available

  cla signed
  #1415·fedebongio opened Mar 25, 2026·

  ·
  3

• Fix iterator exhaustion in language_translation example

  cla signed
  #1412·adityap-codes07 opened Mar 3, 2026·

  ·
  1

  3

• mnist: avoid concurrent MNIST downloads under mp.spawn

  cla signed
  #1411·lordaarush opened Jan 14, 2026·

  ·
  1

  3

• Add optional SyncBatchNorm support to ImageNet example

  cla signed
  #1406·snehasonkusare-tech opened Dec 8, 2025·

  ·
  4

• Refactor(examples/mnist): Clarify loss reduction and correct dataset length

  cla signed
  #1396·sanshi00000030 opened Oct 21, 2025·

  ·
  1

  1

• Adding torch accelerator to ddp-tutorial-series example

  cla signed
  #1393·dggaytan opened Sep 10, 2025·

  ·
  2

• Use of torch.inference_mode() instead of torch.no_grad() for inference

  cla signed
  #1391·thomastschinkel opened Sep 3, 2025·

  ·
  3

• Implement DCGAN example with distributed support (DDP)

  cla signed
  #1390·jafraustro opened Aug 27, 2025·

  ·
  1

• mnist_forward_forward: Load dataset from comment path

  cla signed
  #1389·Rtoax opened Aug 27, 2025·

  ·
  2

• mnist: support --model-path and --load-model

  cla signed
  #1388·Rtoax opened Aug 26, 2025·

  ·
  4

• pass compilation; fail linking to lib/libc10d.a

  #1341·davidmlw opened May 6, 2025·

  ·
  4

• Better device handling

  cla signed
  #1301·EIFY opened Dec 1, 2024·

  ·
  2

• Updated regression example inline with other examples

  cla signed
  #1300·boringbyte opened Nov 14, 2024·

  ·
  1

• Fix platform type inconsistency custom-dataset.cpp

  cla signed
  #1293·nccrrv opened Oct 16, 2024·

  ·
  1

  1

• Check for getting ILSVRC2012 datasets from image-net.org

  cla signed
  #1291·yskzalloc opened Oct 3, 2024·

  ·
  1

• Replace hardcoded default values in argparse help strings

  cla signed
  #1290·miguelsousa opened Sep 26, 2024·

  ·
  2

• Changes to the code

  cla signed
  #1283·doshi-kevin opened Aug 27, 2024·

  ·
  2

• added FSDP MNIST example

  cla signed
  #1276·surprisedPikachu007 opened Jul 18, 2024·

  ·
  1