岗位描述
岗位描述：作为团队核心成员，你将负责面向 AMD GPU 优化和开发深度学习框架，提升 GPU 内核性能，加速深度学习模型，并支持多 GPU 与多节点系统下的大规模训练和推理。你将与内部 GPU 软件团队紧密协作，并参与开源社区合作，集成前沿编译器技术，推动上游贡献，完善 AMD AI 软件生态。具体包括：优化 TensorFlow、PyTorch、SGLang 等框架在 AMD GPU 上的性能；对大规模训练和推理模型进行性能分析与调优；使用 HIP、Triton 等工具设计、实现并优化高性能 GPU Kernel，提高 AI 算子效率；与编译器团队和 GPU 数学库团队协作，推动内核级优化与全栈性能目标对齐；支持 SGLang LLM 框架在 AMD GPU 平台上的优化、功能开发与扩展；优化多 GPU 和多节点环境下的分布式训练与推理性能，包括推理并行和集合通信；集成并优化 XLA、TorchDynamo 或自定义图编译流水线的运行时执行；与外部维护者合作，理解框架需求并推动优化方案上游落地；在调试、性能分析、测试驱动开发和 CI 集成中落实现代软件工程最佳实践。 岗位要求：具备扎实的 Linux 环境下 C++ 开发能力和较强的技术分析能力，能够在团队协作和独立工作场景中高效推进任务，具备目标定义、开发管理和高质量交付能力，拥有较强的问题解决能力、主动性和软件工程最佳实践意识。需要具备 C++ 和/或 Python 编程经验，能够调试、分析和优化性能关键代码；有 SGLang 或类似 LLM 推理框架实操经验者优先；具备编译器设计背景或熟悉 LLVM、MLIR、ROCm 者优先；具备在大规模异构集群（CPU+GPU）上运行和扩展分布式训练或推理工作负载的经验；有将优化集成到 PyTorch、TensorFlow 等深度学习框架中的经验者优先；熟悉 HIP、CUDA 或其他 GPU 编程模型，了解 GCN/CDNA 架构更佳。学历要求为计算机科学、计算机工程、电气工程或相关专业本科及以上学历。