岗位描述(节选)
你将加入PAI平台团队,负责面向各类AI应用场景的模型训练与推理性能优化工作。该岗位聚焦于AI计算性能与系统效率的极致提升,以多层次优化手段,持续提升模型训练与推理的速度、稳定性与资源利用率,具体工作内容包含但不限于:
● 分析客户模型在不同硬件(GPU、NPU、CPU 等)和集群环境中的性能瓶颈,制定优化方案;
● 对主流深度学习框架(如 PyTorch、TensorFlow、JAX 等)进行算子、内核或图优化;
● 深入理解分布式训练架构(如数据并行、模型并行、流水并行等),优化通信与调度性能;
● 在推理端负责模型性能瓶颈分析,并进行量化、剪枝、融合、TensorRT/ONNX Runtime 等优化;
● 与平台团队协作,优化训练任务调度、算力利用、容器化运行效率以及多租户资源隔离;
● 支持重点客户模型性能调优,提供端到端性能优化方案与技术支持。
【任职要求】
必备条件:
● ……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。