研发实习生(AI system)

薪资面议
2025-03-31










职位描述
1.参与大模型在多元芯片、编译器、算子库、框架上的研究与应用工作;
2.对大模型应用场景从功能性、性能、易用性等角度提出系统层面的设计并落地。
任职要求
1.熟悉PyTorch,了解CUDA的定位和基本概念,了解或能快速上手学习Megatron或DeepSpeed的使用方式;
2.了解大模型训练、对齐、推理的相关基本概念;
3.了解linux开发环境,能够快速学习cudac、python以及其他可能用到的新开发语言;
4.认真严谨,有良好的沟通能力;善于学习新知识,有进取心;
5.实习时间要求:6个月(以科研任务为导向,计划长时间培养)。
加分项:
1.了解DeepSeek、Qwen等热门大模型的区别与共性,了解或能快速上手学习RoPE,MLA等算法结构;
2.了解Disaggregated Prefill&Decode,Quantization等性能优化方案;
3.了解TensorCore、HBM、nvlink、IB/RoCE等硬件模块的基本功能效果与使用方式;
4.对一种非英伟达AI芯片有了解,有模型迁移相关经验。
备注:由于该岗位的
工作内容
相对新颖,涉及行业前沿知识,我们欢迎对AI系统方向有兴趣的朋友申请,即使缺乏经验也可以考虑。