Pytorch框架研发高级工程师（深圳/北京/上海/杭州）

岗位职责

1. 参与开发优化pytorch基础框架，支持公司内业务通用训推工程需求；
2. 联合业务进行模型训推效率分析优化、精度对齐、效果验证等工作；
3. 参与基础框架的国产卡适配接入，提升框架异构算力使用效率。

任职要求

1. 熟悉分布式训推框架，精通集合通信、并行策略等基础原理，有 LLM 训练/推理性能优化成功案例者优先；
2. 深入理解时序分析方法，具备异常检测相关能力，有大规模基础设施（IDC、云计算）智能诊断经验者优先；
3. 熟悉常见的任务/故障/性能衰减等问题排查流程，有监控/鲁棒性平台研究经验优先。

产品/项目

大数据(数据计算平台部)

工作年限

一年以上工作经验