AI基础软件开发

AI基础软件开发

Base：杭州/上海

经验要求：应届/社招

学历要求：本科及以上

职责描述：

1.负责工具链项目(模型训练/推理/量化)研发工作，包括模型评估工具，模型精度调优工具等

2.设计面向AI芯片的程序调优，性能分析工具，构建覆盖算子、模型到应用级别的全流程 Profiling 体系

3.优化深度学习训练代码：分析和优化现有PyTorch代码，减少计算冗余，提高GPU资源利用率

4.编写自动化测试与性能 Benchmark 工具，保障模型与工具链的功能正确性、性能表现和长期稳定性

5.与算法工程师协作，参与模型轻量化（量化 / 剪枝 / 蒸馏）后的部署验证，根据算法和业务场景设计测试数据集并跟进问题闭环

6.推动工具链与 CI/CD 流程深度融合，对集成模型进行持续的、性能回归、效果评估与自动化优化

任职要求:

1.具备良好的工程能力，熟练使用Python或C++优化深度学习代码，具备调优大规模AI模型的经验

2.有AI领域相关测试经验，掌握模型量化，推理，训练测试能力者优先

3.具备异构计算平台下工具链研发经验者优先，AI模型训练和推理过程的性能分析、调试和调优工具技术原理和核心流程

4.快速的学习能力，良好的沟通表达能力，良好的分析解决问题能力，有较好的独立思考判断能力

5.了解PyTorch Profiler，Nsight System，py-spy等工具