岗位描述
岗位描述：负责 AMD 生态内各类 AI 框架的开发与技术引领，在 AMD GPU 上开发和优化深度学习框架；与内部 GPU 库团队及开源维护者合作，确保优化方案顺利集成，并利用前沿编译器技术和先进工程方法持续改进；在 AI/ML 框架（如 PyTorch、vLLM、SGLang）、AI 运行时组件及优化工具等核心方向开展工作，加速 AI/ML 负载在 AMD GPU 上运行；与 AI 研究人员紧密合作，推动框架组件开发，使 AI 模型能够高效映射并运行在最新 AMD GPU 上；在开放源码仓库中增强和优化 PyTorch、vLLM、SGLang 等深度学习框架；开发和优化 GPU Kernel，以提升特定 AI 算子的性能；通过量化等手段设计并优化深度学习模型以提升 AMD GPU 性能；与 GPU 库团队协作分析并改进 AMD GPU 上的训练和推理性能；与开源框架维护者协作，确保代码改动符合要求并成功回传上游；遵循良好的软件工程实践，构建稳健且可维护的解决方案。 岗位要求：具备在 AMD GPU 上使用 HIP、CUDA 和汇编（ASM）进行深度学习 GPU Kernel 设计与优化经验；深入理解 AMD 架构（如 GCN、RDNA）和底层编程，能够利用 Compute Kernel（CK）、CUTLASS、Triton 等工具提升多 GPU 和多平台 AI 性能；具备 AI 软件框架经验，如 PyTorch、vLLM、SGLang，以及性能基准测试和性能分析经验；具备大型模型分析与基准测试工具使用经验；具备模型优化经验，如低精度量化（MXFP4、FP8、INT4）和稀疏化；深入理解模型架构、LLM、MoE、扩散模型；精通 C++ 编程；具备 Python 开发与调试经验；具备团队合作精神，能够与分布式全球团队协同工作；计算机科学、计算机工程、电子工程或相关技术领域本科、硕士或博士学历。