示例：https://huggingface.co/spaces/nanotron/ultrascale-playbook?section=splitting_layers_on_various_nodes_-_all_forward%2C_all_backward
