Experimental validation of scaling PP/TP for very large models and CP for very long sequences
Experimentally validate the expected benefits of further increasing pipeline or tensor parallelism for extremely large models and increasing context parallelism for very long sequences, quantifying their impact on Model FLOPs Utilization, throughput, and memory usage in large-scale settings.
References
Further increases in the degree of model parallelism (PP/TP) are expected to benefit extremely large models. Increasing context parallelism (CP) is expected to be advantageous for very long sequences; however, due to resource limitations, experimental validation of these scenarios is left for future work.
Nevertheless, determining the optimal hybrid parallelization strategy for a given training scenario remains an open research problem.
Finally, our benchmarks cover a representative but not exhaustive set of QKV tensor configurations; a systematic sweep over batch sizes and query/key-value head ratios beyond those reported in \S\ref{sec:exp-cp-scalability} and \S\ref{sec:exp-e2e} is left to future work.