高性能算子开发工程师（自研芯片方向）（J97205）

岗位职责

-分析常见大模型在GPU和自研芯片上的性能瓶颈，挖掘自研芯片优化潜力 -基于百度自研芯片，开发常见大模型算子，充分发挥算力极限 -针对常见训推引擎，封装并开发算子API，发挥端到端性能表现 -承接公司内外前沿模型的算子开发需求，扫清模型的算子依赖 -探索利用AI Agent开发或优化GPU和自研芯片算子性能

任职要求

-计算机相关专业，有C/C++/Python基础，熟悉Linux开发环境 -有深度学习框架（PyTorch, TensorFlow, PaddlePaddle等）使用经验优先 -熟悉CUDA、OpenCL、AscendC等高性能计算编程经验优先 -理解量化算子技术，熟悉不同精度数据格式（FP8、FP16、BF16等）优先 -理解卡间、机间通信原理，熟悉模型并行技术（TP、PP、DP等）优先 -理解GPU、XPU、TPU、NPU等AI计算加速芯片架构优先 -理解当代大语言模型架构（Qwen、DeepSeek等）优先 -对AI技术有热情，具备持续学习能力

部门/组织

训推引擎组