megatron.core.parallel\_state.initialize\_model\_parallel ========================================================= .. currentmodule:: megatron.core.parallel_state .. autofunction:: initialize_model_parallel