高级机器学习工程师
岗位摘要
通过模型、数据、工具或其他可用手段持续提升搜索质量;设计和构建搜索平台及模型堆栈的核心组件;使用PyTorch等框架设计、训练和优化大规模深度学习模型,利用分布式训练(如PyTorch Distributed、DeepSpeed、FSDP)和硬件加速,专注于检索和排名模型
岗位职责
- 通过模型、数据、工具或其他可用手段持续提升搜索质量
- 设计和构建搜索平台及模型堆栈的核心组件
- 使用PyTorch等框架设计、训练和优化大规模深度学习模型,利用分布式训练(如PyTorch Distributed、DeepSpeed、FSDP)和硬件加速,专注于检索和排名模型
- 在表示学习领域进行高级研究,包括对比学习、多语言和多模态建模,用于搜索和检索
- 以可扩展和高性能的方式部署模型,从提升算法到大型语言模型(LLMs)
- 构建和优化RAG(检索增强生成)管道,用于基础信息生成和答案生成
- 与数据、人工智能、基础设施和产品团队合作,确保快速高质量的交付
任职要求
- 深入了解搜索和检索系统,包括质量评估原则和指标
- 在大规模搜索或推荐系统方面有良好的记录
- 熟练掌握PyTorch,包括分布式训练技术和大型模型性能优化的经验
- 在表示学习方面有专长,包括对比学习和多语言、多模态应用的嵌入空间对齐
- 在AI/ML会议或研讨会(如NeurIPS、ICML、ICLR、ACL、CVPR、SIGIR)上有较强的发表记录
- 自我驱动,具有强烈的责任感和执行力
- 至少有3年(最好是5年以上)从事搜索、推荐系统或密切相关研究领域的工作经验