1. 25 Feb, 2021 1 commit
  2. 23 Feb, 2021 2 commits
  3. 19 Feb, 2021 1 commit
  4. 25 Jan, 2021 1 commit
  5. 28 Dec, 2020 1 commit
  6. 16 Dec, 2020 1 commit
    • Min Xu's avatar
      [feat]: AdaScale work with lr_scheduler and tests, examples (#229) · d65cd838
      Min Xu authored
      * [doc]: AdaScale example and notes
      
      * formatted notes correctly as suggested by Benjamin
      
      * added feature and unit test to make sure lr_scheduler works
      
      * update the example with lr_scheduler
      
      * fixed doc with "make html"
      
      * addressed Mike's suggestions
      d65cd838
  7. 09 Dec, 2020 1 commit
  8. 05 Dec, 2020 1 commit
  9. 04 Dec, 2020 1 commit
  10. 01 Dec, 2020 1 commit
  11. 27 Nov, 2020 1 commit
  12. 21 Nov, 2020 1 commit
    • Benjamin Lefaudeux's avatar
      [feat] ShardedDataParallel with autoreduce (#157) · ad933b34
      Benjamin Lefaudeux authored
      * rewrite using autograd and Variable execution queue to make the reduce automatic
      * share buckets with OSS to remove duplication
      * some speed still likely on the table since the speed vs. bucketing does not match expectations, could be a follow up
      ad933b34
  13. 18 Nov, 2020 1 commit
  14. 17 Nov, 2020 1 commit
    • Min Xu's avatar
      [doc] add AdaScale API doc (#191) · 587b707d
      Min Xu authored
      - removed experimental warning as we have validated it on cifar and
      imagenet, transformer is looking good so far too.
      - fixed API doc formatting
      - make it consistent with the other code in the repo
      - tested by making the doc locally and inspect the results
      587b707d
  15. 10 Nov, 2020 1 commit
    • Tom Birch's avatar
      Single-process control via PipeRPCWrapper (#156) · 5d4f50fb
      Tom Birch authored
      Adds support for:
      * Reused layers (e.g. for weight sharing)
      * Lazily-constructed layers
      * Single-process control via PipeRPCWrapper
      * PipelineStyle.AsyncScheudle, which lays the foundation for asynchronous pipeline work by introducing an event loop for each rank/worker to process either activations or gradients as they arrive
      
      Also added examples for multi-process and PipeRPCWrapper
      5d4f50fb
  16. 30 Oct, 2020 1 commit
  17. 22 Oct, 2020 2 commits
  18. 21 Oct, 2020 4 commits
  19. 18 Oct, 2020 1 commit
  20. 16 Oct, 2020 1 commit
  21. 10 Oct, 2020 1 commit
  22. 09 Oct, 2020 1 commit
  23. 24 Sep, 2020 1 commit
  24. 22 Sep, 2020 2 commits
  25. 17 Sep, 2020 2 commits
    • Tom Birch's avatar
      Multi-process pipe (#90) · 63f7796a
      Tom Birch authored
      Adds support for distributing pipeline stages across multiple processes (and therefore multiple machines)
      * Adds a style argument to the Pipe constructor, defaulting to PipelineStyle.SingleProcess, but also supporting PipelineStyle.MultiProcess
      * Added support for lazy construction of modules (see lazy_construction for an example)
      * Added two implementations of inter-process communication: one based on rpc with globally visible queues, one based on send/recv
      * Copied all the relevant tests from tests/pipe to tests/pipe_process and modified them to exercise PipelineStyle.MultiProcess
      63f7796a
    • Benjamin Lefaudeux's avatar
      [chore] OSS: add a small sphinx tutorial, similar to README (#92) · 2d415f30
      Benjamin Lefaudeux authored
      Add a small tutorial, similar to the OSS README
      2d415f30
  26. 16 Sep, 2020 1 commit
  27. 12 Sep, 2020 1 commit
  28. 11 Sep, 2020 1 commit
  29. 10 Sep, 2020 2 commits
  30. 09 Sep, 2020 3 commits