岗位描述
岗位职责:
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. 有较好的学习能力和团队合作能力;