AI软件开发工程师(北京)-25届春招

薪资面议
北京市海淀区知春路7号致真大厦D座16层
2025-03-13










岗位职责
1、参与大模型训练框架的适配、优化和开发工作,分析优化性能热点;
2、调研大模型最新技术和性能优化方案,跟进社区演进方向,持续优化大模型训练瓶颈问题;
3、性能分析的理论计算及预估,对计算和通信时间占比能够进行合理性分析,并提出优化方案。
岗位要求
1、硕士及以上学历,微电子、计算机、通信、自动化、应用数学、物理等相关专业;
2、熟悉pytorch的使用,熟悉大模型相关的训练方法和技巧;
3、熟悉Linux/Unix系统环境,能够熟练使用C++/Python;
4、熟悉计算机体系结构,熟悉操作系统概念;
5、思维缜密、执行力强、抗压能力强;
6、良好的沟通能力和团队协作能力,良好的学习能力和自驱能力。
拥有以下经验者优先:
1、有大模型训练或AI框架开发优化经验;
2、有Nsight System,DL Prof等性能分析工具使用经验。