直招.cv

← 返回职位列表

小红书 · AI 工程/Infra 校园招聘

【2027校招】Dots-大模型高性能计算工程师- AI infra

北京 ·经验不限·学历不限

北京市·上海市

需本科及以上学历,计算机相关专业优先,精通GPU CUDA编程及Megatron/DeepSpeed/FSDP框架

GPU CUDA编程MegatronDeepSpeedFSDP模型并行分布式技术算子优化TritonMLSys

职位描述

大模型AI Infrastructure团队专注于大语言模型领域的前沿技术研究和落地,提供高性能、高可靠、可扩展的机器学习系统、丰富的异构计算资源和极致的端到端的机器学习服务体验,为公司提供核心技术能力和服务。 1、负责机器学习框架的研究与开发,服务于公司各个产品; 2、高效部署,优化NLP/多模态大模型核心业务模型。 【任职要求】 1、本科及以上学历,计算机等相关专业方向优先; 2、熟练掌握GPU CUDA编程; 3、熟练掌握Megatron/DeepSpeed/FSDP等框架的研发,熟练掌握模型并行分布式技术; 4、对大语言模型算法感兴趣,有机器学习算法知识背景; 5、追求技术极致,务实,渴望有自己的作品和代表作。 【加分项】 1、对于算子优化(CUDA or Triton)、通信优化有深入钻研者; 2、对芯片体系结构和指令执行pipeline有深刻认知者; 3、MLSys方向的知名开源项目贡献者; 4、ACM等编程竞赛获奖者。

官网发布:2026-09-06 · 最后确认在招:2026-09-07 20:27:27 · 来源平台:xiaohongshu