职位描述 负责大模型基础训推性能优化，聚焦异构GPU的高性能算子库、通信库的开发和优化，提升计算、通信的并行效率，设计并实现高效的并行计算策略、分布式推理方案等。提炼核心的优化方法，探索通用的编译优化方案，跟进前沿技术，并将优化能力集成到自研/开源大模型推理框架，推动优化方案在实际业务场景中的落地，持续创新构建业界领先的AI Infra。

职位要求 1、计算机相关专业本科及以上学历，熟悉C++、Python，具备良好的算法与数据结构基础，有高性能计算相关经验优先； 2、熟悉GPU硬件架构，掌握CUDA/ROCm/Triton/CUTLASS/Composable Kernel等并行编程语言，具备GEMM/FlashAttention等计算Kernel优化经验，或DeepEP等通信Kernel优化经验，熟悉NCCL、NVSHMEM等通信库； 3、熟悉TVM/MLIR/XLA等编译器，具备开发优化Pass的能力； 4、熟悉LLM、Stable Diffusion等主流模型结构，具备研发、调优或部署经验。熟悉vLLM、SGLang等推理框架； 5、了解PyTorch/TensorFlow等深度学习框架。