岗位描述(节选)
岗位职责:
1、国产GPU平台AI模型移植与部署:负责主流AI模型在国产GPU上的端到端迁移、适配与部署,涵盖训练与推理阶段。
2、性能分析与瓶颈诊断:使用Profiling工具进行系统级性能分析,定位瓶颈并提出优化方案,涵盖计算、存储、通信等多维度。
3、GPU算子与计算图优化:基于ROCm/HIP栈进行模型性能优化,包括自定义HIP算子开发与调优;使用CUTLASS、Triton等工具进行算子级优化;计算图融合,内存布局优化等。
4、分布式训练与推理优化:深入理解并应用DP/TP/EP等并行策略,优化大语言模型在国产GPU集群上的分布式训练与推理方案。
5、客户支持与技术方案定制:支持客户AI业务在国产GPU上的落地,解决训练/推理中的性能、精度、兼容性等问题,提供定制化优化方案和技术选型建议。
任职要求:
学历:本科及以上学历,计算机、人工智能、电子信息、高性能计算等相关专业。
经验……
完整岗位描述与官方投递入口,请登录后在岗位详情页查看。
本站当前在招 24807 个上海岗位(全城各职能合计),其中 1207 个公开标注了薪资。按这些标注统计:月薪(税前)主流区间 8K ~ 20K(25~75 分位),中位约 12K,最低 0.8K、最高 113K。
在招岗位较多的企业:字节跳动 958 个米哈游 691 个拼多多 497 个哔哩哔哩股份有限公司 452 个阿里巴巴集团 442 个
统计口径:求职者营地实时在招岗位的公开薪资标注,未标注者不计入;区间为 25~75 分位,非薪酬承诺。