小红书 校园招聘
【REDstar】Dots-大模型高性能计算工程师- AI infra
北京 ·经验不限·学历不限
北京市·上海市
薪资面议
前往 小红书 官网投递
职位描述
大模型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-02 · 最后确认在招:2026-09-03 23:16:32 · 来源平台:xiaohongshu