计算框架工程师

岗位职责: 1. 基于自研GPGPU适配、维护常用计算框架，如PyTorch、TensorFlow(XLA)、Megatron-LM等； 2. 实现或优化计算框架分布式训练，提升模型训练性能； 3. 优化或重写计算框架的算子以达到最优的性能； 4. 分析、定位、解决模型训练或推理过程中涉及的计算框架问题。 任职要求： 1. 具有计算机相关领域背景，熟悉AI领域基本知识； 2. 熟练使用Python/C++编程，熟悉CUDA编程； 3. 熟悉PyTorch或TensorFlow(XLA)等框架； 4. 熟悉DP、TP、PP、CP等并行策略，熟悉通信库NCCL； 5. 熟悉Transformer、GPT主流模型结构，熟悉NLP常用算法原理； 6. 有AI框架、算子开发、通信库开发经验者优先； 7. 有大模型Pre-train、Fine-tune调优经验者优先。