`bridge.perf_recipes.kimi.vrnvl72.kimi_k2`#

VRNVL72 performance recipes for Kimi K2.

Module Contents#

Functions#

`kimi_k2_pretrain_256gpu_vrnvl72_bf16_config`	Kimi K2 pretrain: 256× VRNVL72, BF16 (alias of GB300).
`kimi_k2_pretrain_256gpu_vrnvl72_fp8mx_config`	Kimi K2 pretrain: 256× VRNVL72, MXFP8 (alias of GB300).

API#

bridge.perf_recipes.kimi.vrnvl72.kimi_k2.kimi_k2_pretrain_256gpu_vrnvl72_bf16_config() → megatron.bridge.perf_recipes.kimi.common.ConfigContainer#: Kimi K2 pretrain: 256× VRNVL72, BF16 (alias of GB300).

bridge.perf_recipes.kimi.vrnvl72.kimi_k2.kimi_k2_pretrain_256gpu_vrnvl72_fp8mx_config() → megatron.bridge.perf_recipes.kimi.common.ConfigContainer#: Kimi K2 pretrain: 256× VRNVL72, MXFP8 (alias of GB300).