2027届校招机器学习平台研发工程师-上海

AI 摘要智能生成

参与深度学习模型计算图拆分与并行化工具链的设计与

职位描述

>1、参与深度学习模型计算图拆分与并行化工具链的设计与开发，提升模型在多核/多设备环境下的部署效率； >2、负责基于 ONNX 的算子切分与重写规则库的建设、扩展与长期维护； >3、针对模型并行部署场景，分析并解决算子精度对齐、Shape 兼容性、边界条件等工程问题； >4、协同算法、编译器与硬件团队，推进模型从训练框架到推理平台的稳定落地； >

任职要求

• >
• 计算机、电子工程、自动化或相关专业，本科及以上学历；
• 熟练掌握 Python，具备良好的工程代码规范与调试能力；
• 熟悉 ONNX 模型格式、计算图结构，理解算子拓扑关系与数据依赖；
• 熟悉 PyTorch / TensorFlow 等主流深度学习框架，具备模型转换或部署经验；
• 对底层技术有较强兴趣，具备主动学习和持续深入钻研的能力。