Infra推理优化实习生(3D生成)当面告知

上海米哈游网络科技股份有限公司
2027-02-05 16:47:31
博士28届29届实习硕士

岗位详情

地区:

上海

专业要求:

人工智能计算机科学与技术计算机类

岗位职责:

1、支撑团队不同大模型任务下的训练及推理优化,包括但不限于:架构设计、检查点优化、训练容错恢复、底层算子优化及通信优化等; 2、研究和优化大规模异构加速集群调度、存储、通信互联、监控、Profiling等组件,提升整体硬件利用效率; 3、针对具体任务调研并进行对应的 Scale Up 实验及优化,包括但不限于并行化策略优化、ZeRO/FSDP 优化、FP8、激活值优化等,加速模型训练,优化显存开销;

招聘要求:

1、硕士及以上学历,具有扎实的计算机基础,熟悉操作系统和计算机体系结构等,熟练掌握 C++/Python,对数据结构与算法设计有较深刻的理解; 2、熟悉至少一种主流的深度学习框架(如 PyTorch)和至少一种主流的训练框架(如 Megatron/Deepspeed/FSDP); 3、熟练掌握 CUDA 相关编程,有内核级优化经验者优先; 4、了解大模型训练基本范式,熟悉 Transformer 架构及其优化方法(如 FlashAttention、GQA/MLA 等)者优先; 5、责任心强,思路清晰,技术视野开阔,对业界新技术敏感,喜欢钻研,具有良好的学习能力并注重团队合作; 加分项: 1、有大规模大模型训练和推理优化经验者优先; 2、熟悉cutlass 3.0和hopper架构,有算子优化经验者优先; 3、了解 Kubernetes 以及分布式存储,有相关开发经验者优先; 4、有大型互联网系统(如搜索、推荐、广告)开发经验者优先; 5、有发表 OSDI/MLSys 等顶会论文,或 ACM/ICPC 等编程比赛获奖经历者优先;

公司介绍

上

上海米哈游网络科技股份有限公司

上海米哈游网络科技股份有限公司,成立于2012-02-13,位于上海,法定代表人:刘伟,登记状态:存续,注册资本31716.6663万人民币,公司类型:民营/私企
公司信息:
公司名称:上海米哈游网络科技股份有限公司
公司类型:网络游戏
公司性质:1000-4999人
公司地点:
公司规模:--
涉及领域:--
公司网址: