megatron.initialize.set_jit_fusion_options# megatron.initialize.set_jit_fusion_options(args)# Set PyTorch JIT layer fusion options.