AI框架及编译技术专家(Pytorch&Triton)当面告知

沐曦集成电路(上海)股份有限公司
2026-11-15 22:33:48
其他社招

岗位详情

地区:

上海市
·
杭州市

专业要求:

人工智能计算机科学与技术软件工程

岗位职责:

1. 负责PyTorch在沐曦GPU上的适配和优化,跟踪并引入PyTorch生态下各类新技术; 2. 负责OpenAI Triton在沐曦GPU上的适配和优化,演进Pythonic Programming的编译技术; 3. 支持流行的开源AI训练框架以及客户定制的AI训练框架; 4. 实现高性能算子,提升主流AI模型在GPU上的性能

招聘要求:

1. 熟悉pytorch/triton框架,了解vLLM/SGLang/Megatron-LM; 2. 至少3年C++编程经验; 3. 具有CUDA, ROCM, OpenCL相关开发经验者优先; 4. 优先加分项:具有AI算子开发和优化经验; 5. 优先加分项:具有AI训练框架和AI推理框架交付经验;

公司介绍

沐

沐曦集成电路(上海)股份有限公司

沐曦集成电路(上海)股份有限公司,成立于2020-09-14,位于上海,法定代表人:陈维良,登记状态:存续,注册资本36000万人民币,公司类型:民营/私企