bridge.perf_recipes.gpt_oss.b200.gpt_oss#

B200 performance recipes for GPT-OSS.

Module Contents#

Functions#

gpt_oss_120b_pretrain_64gpu_b200_bf16_config

GPT-OSS 120B pretrain: 64× B200, BF16, GBS=1280.

gpt_oss_120b_pretrain_64gpu_b200_fp8mx_config

GPT-OSS 120B pretrain: 64× B200, FP8-MX.

API#

bridge.perf_recipes.gpt_oss.b200.gpt_oss.gpt_oss_120b_pretrain_64gpu_b200_bf16_config() megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer#

GPT-OSS 120B pretrain: 64× B200, BF16, GBS=1280.

bridge.perf_recipes.gpt_oss.b200.gpt_oss.gpt_oss_120b_pretrain_64gpu_b200_fp8mx_config() megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer#

GPT-OSS 120B pretrain: 64× B200, FP8-MX.