岗位描述(节选)
岗位职责
1、参与深度学习编译器中计算图切分算法、约束传播与搜索策略的设计与优化;
2、面向 Transformer / LLM 等模型,解决分布式推理与并行中的核心技术问题;
3、针对特定RK自研的NPU/AI加速器架构,设计高效的内存规划、算子调度与多核通信策略;
4、分析推理性能瓶颈,从算法、编译与系统层面持续提升端侧/边缘侧推理效率;
岗位要求
1、计算机及相关专业硕士及以上学历,2 年以上机器学习算法或工程开发经验;
2、精通 Python,熟悉 PyTorch 框架及其计算图执行机制;
3、具备扎实的数据结构与算法基础,熟悉 树/图遍历、搜索、约束传播等基本方法;
4、逻辑思维清晰,能够系统性分析并解决复杂技术问题;
5、对高性能计算与模型推理系统有持续投入的技术热情。