Commits · 52a5f2f272e5ef242eb271227da712f7dfc55da3 · OpenDAS / Megatron-LM

12 Nov, 2020 2 commits

Intra-layer MP -> Tensor MP, Inter-layer MP -> Pipeline MP · 52a5f2f2
Deepak Narayanan authored Oct 20, 2020

52a5f2f2

Pipeline parallelism implementation with periodic full-pipeline syncs · 7abd3e90

Deepak Narayanan authored Aug 29, 2020

Also includes following changes for inter-layer model-parallel implementation:
- Refactoring of model implementations
- Training loop changes to support inter-layer communication using `ring_exchange`
- New groups for inter-layer communication
- Checkpoint changes
- Command line arguments

7abd3e90

16 Apr, 2020 4 commits
- Fix finetine -> tune · 5cb2d1ea
  Steven Steinke authored Apr 16, 2020
  
  5cb2d1ea
- fix typos · 50391a0b
  Steven Steinke authored Apr 16, 2020
  
  50391a0b
- updated scripts · ca19a8b5
  Mohammad authored Apr 16, 2020
  
  ca19a8b5
- moved steves branch · 78cb1781
  Mohammad authored Apr 16, 2020
  
  78cb1781