锦鲤求职

沐曦股份

AI框架及编译技术专家(Pytorch&Triton)

锦鲤会替你打开官网网申、按简历自动填表提交,你只需要在关键步骤确认。

岗位描述

职责描述:

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推理框架交付经验;

人工智能方向的申请准备

先核对岗位要求与自己的经历,再准备档案、投递和面试。

阅读求职步骤与示例 →