Instructions to use KexuanShi/Megatron-LM with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NeMo
How to use KexuanShi/Megatron-LM with NeMo:
# tag did not correspond to a valid NeMo domain.
- Notebooks
- Google Colab
- Kaggle
|
Download docs/api-guide/core/pipeline_parallel.md from KexuanShi/Megatron-LM: direct link, hf CLI and curl.
- Browser
- Download file 444 Bytes
-
https://huggingface.co/KexuanShi/Megatron-LM/resolve/main/docs/api-guide/core/pipeline_parallel.md
- Command line
-
hf download hf://KexuanShi/Megatron-LM/docs/api-guide/core/pipeline_parallel.md
-
curl -L -o pipeline_parallel.md https://huggingface.co/KexuanShi/Megatron-LM/resolve/main/docs/api-guide/core/pipeline_parallel.md
444 Bytes
| # pipeline_parallel package | |
| This package contains implementations for two different pipeline parallelism | |
| schedules (one without interleaving and one with interleaving, see [Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LM](https://arxiv.org/abs/2104.04473) | |
| for details), and a default no-pipelining schedule. It also contains methods | |
| for the point-to-point communication that is needed between pipeline stages. | |