bridge.perf_recipes.gpt_oss.gb300.gpt_oss

GB300 performance recipes for GPT-OSS.

Module Contents

Functions

gpt_oss_20b_pretrain_8gpu_gb300_nvfp4_config	GPT-OSS 20B pretrain: 8× GB300, NVFP4.
gpt_oss_20b_pretrain_72gpu_gb300_nvfp4_config	GPT-OSS 20B pretrain: 72× GB300, NVFP4.
gpt_oss_20b_pretrain_512gpu_gb300_fp8mx_config	GPT-OSS 20B pretrain: 512× GB300, MXFP8.
gpt_oss_120b_pretrain_64gpu_gb300_bf16_config	GPT-OSS 120B pretrain: 64× GB300, BF16, GBS=1280.
gpt_oss_120b_pretrain_64gpu_gb300_fp8mx_config	GPT-OSS 120B pretrain: 64× GB300, FP8-MX.

API

bridge.perf_recipes.gpt_oss.gb300.gpt_oss.gpt_oss_20b_pretrain_8gpu_gb300_nvfp4_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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

bridge.perf_recipes.gpt_oss.gb300.gpt_oss.gpt_oss_20b_pretrain_72gpu_gb300_nvfp4_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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

bridge.perf_recipes.gpt_oss.gb300.gpt_oss.gpt_oss_20b_pretrain_512gpu_gb300_fp8mx_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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

bridge.perf_recipes.gpt_oss.gb300.gpt_oss.gpt_oss_120b_pretrain_64gpu_gb300_bf16_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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

bridge.perf_recipes.gpt_oss.gb300.gpt_oss.gpt_oss_120b_pretrain_64gpu_gb300_fp8mx_config() → megatron.bridge.perf_recipes.gpt_oss.common.ConfigContainer

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