Search papers, labs, and topics across Lattice.
This paper addresses the challenges of full-parameter post-training for trillion-parameter MoE models on Ascend SuperPOD, introducing a hierarchical optimization framework that enhances model-level parallelism and computation-communication orchestration. The resulting system achieves a Model FLOPs Utilization (MFU) of 34.22%, marking a 2.93x improvement over existing open-source baselines while ensuring training stability. Furthermore, the integrated framework, SLAI T-Rex, successfully establishes a specialized data pipeline for Operations Research tasks, yielding a model that surpasses GPT-5.4-Mini and the base DeepSeek-V4-Flash in zero-shot performance.
Achieving a 2.93x improvement in training efficiency for trillion-parameter models could redefine the scalability of large-scale AI systems.
Full-parameter post-training of trillion-parameter-scale MoE models introduces substantial system-level challenges for large-scale distributed training, including severe memory pressure, non-overlapped communication overhead, and inefficient kernel execution. While most large-scale LLM training systems are built around GPU-based clusters, this report presents an end-to-end optimization practice on the Ascend NPU SuperPOD. Using the DeepSeek-V4 model family as the target workload, we develop a hierarchical optimization framework spanning model-level parallelism, computation-communication orchestration, and low-level kernel execution. The resulting system achieves 34.22% Model FLOPs Utilization (MFU) with a 2.93x improvement over the open-source baseline recipe while maintaining training stability. Building on this optimized infrastructure, we further establish a CPT and SFT workflow for complex Operations Research (OR) tasks. We refer to the integrated framework as SLAI T-Rex. Using DeepSeek-V4-Flash, we develop OR-oriented CPT and SFT data pipelines that combine collected domain resources with solver-verified synthetic optimization documents. The resulting dataset contains 10K high-quality SFT samples spanning four task categories and three problem representations. The specialized model achieves the highest average zero-shot Pass@1 score among the evaluated models, reaching 71.81% and outperforming GPT-5.4-Mini and the base DeepSeek-V4-Flash model by 3.98 and 11.27 percentage points, respectively. Overall, this work demonstrates a full-stack pathway from efficient trillion-parameter model post-training on Ascend infra to domain-specialized Flash models for solver-grounded mathematical modeling, advancing frontier-model systems for complex reasoning.
