AI算子库性能优化专家(大模型推理MOE layer)-上海/杭州/南京当面告知
沐曦集成电路(上海)股份有限公司
2026-11-15 22:33:50
其他社招
岗位详情
地区:
上海市
·
杭州市
专业要求:
电子科学与技术人工智能计算机科学与技术软件工程
岗位职责:
1. 负责AI算子库在大模型推理背景MOE Layer的核心数学库算子(Fused Moe/Masked Group Gemm/Contiguous Group Gemm/Mega Moe)的多种数据类型实现(bf16/int8/int4/fp8/mxfp4)在沐曦GPU上的功能实现和性能优化;
2. 负责支持热门大模型推理(DeepSeek、GLM、Qwen、MiniMax、Kimi等)中MOE layer在沐曦GPU上的性能优化;
3. 支持流行的大模型推理框架及上层调用,如SGLang/vLLM/DeepGemm等;
招聘要求:
1. 至少3年C++编程经验,具备良好的编程能力和工程经验;
2.了解vLLM/SGLang/DeepGemm;
3. 具有CUDA, ROCM, OpenCL相关开发经验者优先;
4. 具有cutlass相关开发经验者优先
5. 优先加分项:具有AI算子开发和优化经验;
公司介绍
沐
沐曦集成电路(上海)股份有限公司
沐曦集成电路(上海)股份有限公司,成立于2020-09-14,位于上海,法定代表人:陈维良,登记状态:存续,注册资本36000万人民币,公司类型:民营/私企