岗位描述
岗位职责: 1、 参与 主流 AI 框架(PyTorch/JAX/Megatron-LM 等)的二次开发、定制化扩展或生态适配; 2、 参与 AI 计算框架的设计与研发工作,包括数据处理、模型整合、训推流程、强化学习、分布式计算等核心模块; 3、 负责 大规模集群环境下的训练任务部署、问题定位与性能调优; 4、 与算法、平台及算力团队协作,支撑模型稳定、高效运行。 任职要求: 1、 计算机科学与技术、软件工程、自动化、电子信息等相关专业; 2、 具备良好的代码设计能力、调试能力和工程规范意识; 3、 熟练使用 Python,具备良好的工程代码能力,熟悉C++者优先; 4、 熟悉主流 AI 框架的基本原理与使用方式; 5、 有大规模模型训练或集群环境下的问题排查与性能调优经验。 职位类别: 校园招聘