NPU平台高性能计算库开发工程师
岗位摘要
完成高性能算子的需求分析与分解，满足NPU的各种应用场景；与芯片/架构团队紧密配合，持续优化算子性能，逐步完善软硬件架构；完成各类算子的软件方案设计及开发；分析算子/网络的性能瓶颈，提出性能优化策略，达成性能目标
岗位职责
- 完成高性能算子的需求分析与分解，满足NPU的各种应用场景
- 与芯片/架构团队紧密配合，持续优化算子性能，逐步完善软硬件架构
- 完成各类算子的软件方案设计及开发
- 分析算子/网络的性能瓶颈，提出性能优化策略，达成性能目标
任职要求
- 图像处理、模式识别、机器学习等相关专业硕士及以上学历
- 精通C++/Python软件编程语言
- 熟悉Onnx/TensorFLow/Pytorch一种主流神经网络框架，熟悉各类算子
- 熟悉并行/异构计算体系结构，在TPU/NPU/GPU一种平台有2年以上的开发和优化经验
- 具备业界常用高性能库(TensorRT/cuDNN/CUDA/OPENCV)开发和性能优化经验者优先
- 熟悉指令流水，计算机体系结构
- 良好的团队沟通能力、技术攻关能力，责任心强