壁仞科技
算子开发与性能优化工程师/架构师
RESPONSIBILITIES
岗位职责
1. 负责 AI 核心算子(GEMM、卷积、Attention 等)高性能 Kernel 开发与调优
2. 负责 GLM、DeepSeek、Minimax 等大模型推理 / 训练场景的端到端性能分析和协同优化
3. 完成 PyTorch、Megatron、vLLM、SGLang
等训练、推理框架的算子适配与后端集成工作
4. 协同软硬件架构师开展软件流程与架构方案探索,定位性能瓶颈,推动跨团队性能问题落地闭环
##
REQUIREMENTS
任职要求
1. 计算机、电子信息相关专业本科及以上学历,扎实的数据结构与算法基础;精通 C++,熟悉 Python
2. 深入理解 GPU/NPU 体系结构,掌握并行计算与异构高性能编程,具备 CUDA/OpenCL/Triton 开发实战经验
3. 熟悉深度学习基础算法与主流模型,精通算子优化,具备模型训练、推理全链路调优能力
4. 能够熟练使用 Nsight
Compute 等性能分析工具,熟练掌握至少一种主流深度学习框架
## 加分项
1. 拥有 CUTLASS、自定义 CUDA Kernel 开发与优化经验
2. 熟悉 vLLM、SGLang、DeepSpeed、Megatron-LM
等大模型加速、分布式推理 / 训练相关技术
3. 了解 TVM、MLIR、XLA 等 AI 编译器的原理与使用
4. ACM-ICPC 区域赛及以上获奖,或拥有 ISCA/MICRO/PPoPP 等体系结构方向顶会论文
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。
