北极雄芯信息科技(西安)有限公司
AI 算子开发工程师(大模型推理方向)(无锡)
RESPONSIBILITIES
岗位职责
负责大模型推理框架(如 SGLang、vLLM)的底层核心算子开发、移植与性能优化;
配合框架与编译器团队,将自研高性能算子无缝接入 SGLang、vLLM 等推理引擎,提升大模型推理的整体性能;
跟踪大模型推理、高性能计算及 AI 算子优化领域的前沿技术,推动相关技术的工程化落地。;
编写高质量的技术文档与算子使用规范,协助解决推理框架在算子层面的复杂调试与精度对齐问题。
REQUIREMENTS
任职要求
熟练使用 AI 工具,具备运用 AI 工具解决复杂疑难问题的经验。;
计算机、电子工程、微电子、自动化等相关专业本科及以上学历,硕士、博士优先;
编程基础扎实,熟练掌握 C/C++、Python,具备良好的数据结构与算法基础以及代码调试能力;
熟悉 GPU 体系结构及内存层次,熟练掌握 CUDA 编程;或具有 Triton、OpenCL、ROCm 等异构计算开发经验;
熟悉 PyTorch 等主流深度学习框架;了解 vLLM、SGLang、TensorRT-LLM 等大模型推理框架及其底层运行机制者优先;
具备优秀的学习能力、逻辑分析能力和团队协作精神,对大模型底层技术与高性能计算有浓厚兴趣
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。
