职位描述1.协助完成开源模型在国产 GPU/NPU 上的迁移、跑通与精度对齐。 2.在指导下搭建训练与推理环境,复现基线结果,排查算子、通信和显存相关问题。 3.协助性能测试并整理数据,输出 profiling 报告,配合定位瓶颈。 4.维护自动化测试与回归脚本,整理适配文档。任职要求1.本科及以上在读,计算机、人工智能、数学等相关专业;可实习6个月以上,每周到岗3天以上。 2.熟悉 Python,能用 PyTorch 完成训练或推理,了解权重加载、混合精度等基本概念。 3.熟悉 Linux 常用命令,能看懂日志和报错;有 GPU 使用经验即可,国产芯片经验不作要求。 4.学习能力强、做事细致,能主动查资料并写清楚结论。 加分项 1.用过昇腾、寒武纪等国产芯片,或做过模型部署、推理优化。 2.用过 PyTorch Profiler、Nsight 等工具分析过性能。 3.有开源项目、论文复现或竞赛经历。