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