bridge.perf_recipes.gpt_oss.gb200.gpt_oss

GB200 performance recipes for GPT-OSS.

Module Contents

Functions

gpt_oss_20b_pretrain_8gpu_gb200_nvfp4_config	GPT-OSS 20B pretrain: 8× GB200, NVFP4.
gpt_oss_20b_pretrain_72gpu_gb200_nvfp4_config	GPT-OSS 20B pretrain: 72× GB200, NVFP4.
gpt_oss_20b_pretrain_512gpu_gb200_fp8mx_config	GPT-OSS 20B pretrain: 512× GB200, MXFP8.
gpt_oss_120b_pretrain_64gpu_gb200_bf16_config	GPT-OSS 120B pretrain: 64× GB200, BF16, GBS=1280.
gpt_oss_120b_pretrain_64gpu_gb200_fp8mx_config	GPT-OSS 120B pretrain: 64× GB200, FP8-MX.

API

bridge.perf_recipes.gpt_oss.gb200.gpt_oss.gpt_oss_20b_pretrain_8gpu_gb200_nvfp4_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

GPT-OSS 20B pretrain: 8× GB200, NVFP4.

bridge.perf_recipes.gpt_oss.gb200.gpt_oss.gpt_oss_20b_pretrain_72gpu_gb200_nvfp4_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

GPT-OSS 20B pretrain: 72× GB200, NVFP4.

bridge.perf_recipes.gpt_oss.gb200.gpt_oss.gpt_oss_20b_pretrain_512gpu_gb200_fp8mx_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

GPT-OSS 20B pretrain: 512× GB200, MXFP8.

bridge.perf_recipes.gpt_oss.gb200.gpt_oss.gpt_oss_120b_pretrain_64gpu_gb200_bf16_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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

bridge.perf_recipes.gpt_oss.gb200.gpt_oss.gpt_oss_120b_pretrain_64gpu_gb200_fp8mx_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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