岗位描述(节选)
岗位职责
1、参与深度学习模型计算图拆分与并行化工具链的设计与开发,提升模型在多核/多设备环境下的部署效率;
2、负责基于 ONNX 的算子切分与重写规则库的建设、扩展与长期维护;
3、针对模型并行部署场景,分析并解决算子精度对齐、Shape 兼容性、边界条件等工程问题;
4、协同算法、编译器与硬件团队,推进模型从训练框架到推理平台的稳定落地;
岗位要求
1、计算机、电子工程、自动化或相关专业,本科及以上学历;
2、熟练掌握 Python,具备良好的工程代码规范与调试能力;
3、熟悉 ONNX 模型格式、计算图结构,理解算子拓扑关系与数据依赖;
4、熟悉 PyTorch / TensorFlow 等主流深度学习框架,具备模型转换或部署经验;
5、对底层技术有较强兴趣,具备主动学习和持续深入钻研的能力。