岗位描述
工作职责
1、参与设计开发AI 软件栈开发,包括架构设计、推理引擎、异构调度、图优化、算子开发、算子分析性能优化、编译器优化、AI工具集等;
2、AI网络算子开发、调试优化、整网性能调优,包括算子融合、高性能算子库优化并行计算指令流、针对业务场景异构调度最大化资源利用;
3、 负责AI 编译器架构设计开发,负责AI编译器核心模块开发,包括多核、异构计算优化、算子性能调优、模型量化及编译技术研发;
参与芯片架构设计与软硬件协作方案制定,提升编译器与底层硬件的适配性。
4、AI 可视化Studio工具开发、SDK工具链集成与二次开发,AI模型工程部署、可视化图优化、可视化性能Debugging&Profiler;
5、负责大模型的产品化集成落地,满足处理速度、功耗、稳定性、数据集训练等各项指标,负责Android AI 大模型的开发和部署、模型优化。
工作要求
1、计算机、电子、自动化、通信等相关专业硕士及以上学历;
2、较强的编程能力,熟练使用Python、R或Java、C/C++、Halide、Rust等编程语言(有国家证优先);
3、对机器学习和深度学习有深入理解,包括神经网络、回归分析、聚类等算法等,熟悉深度学习算法(CNN/Transformer等),具备算子公式推导能力;
4、熟悉计算机体系结构、并行计算原理,熟悉ARM\NPU\GPU等微处理器架构、Linux/Android系统、多线程编程,并具有良好的代码开发素养;
5、熟悉异构计算编程(如CUDA/OpenCL)及AI框架(TensorFlow/PyTorch);
6、有TVM、MLIR、LLVM等编译器开发经验者优先;
7、熟练掌握TensorFlow、PyTorch、Scikit-learn等工具、能够处理大数据集,熟悉SQL、Hadoop、Spark等数据处理工具;
8、较强的责任心和自我驱动力,良好的沟通能力与合作精神。