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