岗位描述
职位描述:
开发面向大模型需求的高性能算子,优化 NVIDIA 及国产硬件上 kernel 的性能
各硬件平台复杂算子性能分析、极致优化
调研、应用前沿技术,支持开源贡献、论文发表
算子库的集成验证、bug 修复、维护 CI/CD等
任职要求:
本科及以上学历,计算机或相关专业优先;
了解算子开发流程,掌握 triton、tilelang、CUDA、Ascend C 等编程语言之一;
熟悉Python、Shell 等编程语言,具有工程开发经验;
了解 pytorch、vllm、sglang 等深度学习框架;
学习能力强,对技术有热情,有责任心,有良好的团队合作和协调沟通能力。
加分项:
具备高性能算子开发经验优先;
熟悉 AI 芯片硬件架构或 AI 编译器者优先。
了解 pytorch、vllm、sglang 等深度学习框架;
学习能力强,对技术有热情,有责任心,有良好的团队合作和协调沟通能力。
博士优先