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