岗位职责
- 加速LLM新模型架构里的算子和计算图;
- 加速LLM的分布式训练和推理;
- 设计新的算法和框架自动化优化LLM算子,计算图和分布式计算。
职位要求
- AI infra研究背景的博士生,特别优秀的硕士生亦可;
- 在系统相关领域的顶尖学术会议(OSDI,SOSP,EuroSys,VLDB,Sigmod,SC,MLSys等)有AI infra相关的一作论文;
- 有cuda开发经验和并行计算/分布式计算经验;
- 算法竞赛获得优异成绩者优先;
- 有良好的表达和沟通能力及协作精神;工作认真严谨,有强烈的责任心。