megatron.schedules.send_forward_recv_backward#

megatron.schedules.send_forward_recv_backward(output_tensors, tensor_shapes, timers)#