megatron.initialize.set\_jit\_fusion\_options ============================================= .. currentmodule:: megatron.initialize .. autofunction:: set_jit_fusion_options