我见过的其他模型并行API都在模型代码中内联了分区注释，并使用自定义训练逻辑——这使得修改和维护分区配置变得具有挑战性。Keras 3极大地简化了这一过程。