megatron.schedules.send_backward_recv_forward#

megatron.schedules.send_backward_recv_forward(input_tensor_grads, tensor_shapes, timers)#