AI基础软件设计专家(算子)-杭州当面告知

沐曦集成电路(上海)股份有限公司
2027-01-03 22:20:27
社招博士硕士

岗位详情

地区:

杭州市

专业要求:

软件工程

岗位职责:

1. 负责主流 AI 计算框架(如Pytorch、Onnxruntime、Paddle、Tensorflow、MindSpore等)下算子的性能分析、优化策略制定与实施,提升算子在各类计算任务中的执行效率,降低资源消耗。 2. 负责主流LLM推理框架(如vLLM、sglang等)下算子的性能分析与优化。 3. 负责主流训练框架(如Megatron、DeepSpeed等)下算子的性能分析与优化。 4. 负责各种AI相关库的适配与算子性能优化,比如Open-CV、CV-cuda、MMCV、TritonServer等。 5. 跟踪 AI 技术发展趋势,引入新的算子优化技术和工具,不断提升团队的算子分析与优化能力。

招聘要求:

1. 获得计算机、电子工程、软件工程等相关专业硕士以上学位。 2. 熟悉AI主流计算框架及算子库实现,有相关开发经验者优先。 3. 有Cuda或者triton算子优化经验者优先。 4. 熟悉tensorRT并有深入分析者优先。 5. 熟悉C++和Python编程,至少2年以上C++编程经验。 6. 有较好的学习能力和团队合作能力;

公司介绍

沐

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

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