岗位描述
岗位职责
1. 支持端到端模型导出适配和部署加速;
2. 负责支持模型前后处理的板端算法实现和加速;
3. 负责算法模块全流程的性能profiling,针对热点函数实现板端异构加速;
4. 负责模型分布式训练加速,满足自动驾驶模型快速迭代。
岗位要求
1. 硕士及以上学历,计算机相关专业,三年以上相关工作经验;
2. 熟练掌握C++编程,熟悉常用linux工具,有良好的编程习惯;
3. 深入理解pytorch等神经网络模型训练过程,熟悉常用分布式训练加速框架,掌握训练加速调优手段;
4. 了解主要算子的量化方法,熟悉PTQ/QAT 进行加速推理的流程;
5. 熟悉cuda并行加速和cuda算子开发;
6. 英伟达、地平线等AI芯片平台有模型部署经验者优先;
7. 良好的学习能力和强烈的责任心,良好的沟通和团队合作精神。