logo of xiaohongshu

小红书搜广推 - 训练框架研发工程师 / 专家

社招全职引擎地点:北京 | 上海状态:招聘

任职要求


职位要求:
1、熟练掌握Linux环境下的C/C++Python语言;
2、接触过至少一种机器学习框架(Tensorflow / PyTorch / MxNet 或其他自研框架);
3、有以下至少一项的背景知识与经验:GPU编程,编译器,高性能网络,分布式存储,集群调度;
4、具有独立解决问题的能力,良好的团队合作精神;
5、有强烈的工作责任心,较好的学习能力、沟通能力和自驱力;
6、有良好的工作文档习惯,及时按要求撰写更新工作流程及技术文档。

加分项:
1、深入研究过至少一种机器学习框架(Tensorflow / PyTorch / MxNet 或其他自研框架)的底层架构和机制;
2、熟悉至少一种经典深度学习模型及其应用场景,如CTR模型、ResNet50,BERT,或者了解GAN,强化学习,图神经网络,AutoML等;
3、熟悉CUDA编程和NV GPU架构或者其他XPU架构以及编程开发;

工作职责


【业务介绍】
作为公司统一的模型引擎团队,支撑公司内所有搜推广类业务的工程侧工作,包括了模型推理、训练、参数服务器、特征工程等服务,通过引擎能力的持续建设结合多元异构算力为业务提供高效、灵活、稳定的模型服务。
【岗位职责】
1、负责小红书搜广推-机器学习训练框架的研究与开发,服务于全公司各个产品;
2、参与机器学习训练框架底层组件的抽象,设计,优化与落地;
3、与全公司算法部门深度合作,为重点项目进行算法与系统的联合优化。
包括英文材料
Linux+
C+
C+++
Python+
机器学习+
TensorFlow+
PyTorch+
深度学习+
BERT+
强化学习+
CUDA+
相关职位

logo of xiaohongshu
社招3-5年引擎

【业务介绍】 作为公司统一的模型引擎团队,支撑公司内所有搜推广类业务的工程侧工作,包括了模型推理、训练、参数服务器、特征工程等服务,通过引擎能力的持续建设结合多元异构算力为业务提供高效、灵活、稳定的模型服务。 【岗位职责】 1、负责小红书搜广推-机器学习训练框架的研究与开发,服务于全公司各个产品; 2、参与机器学习训练框架底层组件的抽象,设计,优化与落地; 3、与全公司算法部门深度合作,为重点项目进行算法与系统的联合优化。

更新于 2025-10-18
logo of xiaohongshu
社招基础后端

岗位职责: 1、负责小红书搜广推-机器学习训练框架的研究与开发,服务于全公司各个产品; 2、参与机器学习训练框架底层组件的抽象,设计,优化与落地; 3、与全公司算法部门深度合作,为重点项目进行算法与系统的联合优化。

logo of dewu
社招3年以上技术类

团队负责研发一站式搜广推模型机器学习平台,该岗位专注于面向搜索/推荐/广告场景的机器学习分布式训练系统研发 1、负责研发搜广推稀疏大模型分布式GPU同步训练框架,支持GPU单机多卡、多机多卡训练,支持稀疏参数的多级缓存架构和异步流水训练,满足搜广推稀疏大模型离线训练、在线学习、特征准入和淘汰等算法需求; 2、负责TensorFlow/Pytorch框架的后端GPU训练性能优化,个别自定义GPU算子的cuda kernel开发和优化; 3、参与样本数据平台、搜推模型全链路解决方案、LLM4REC等重要方向的研发,提升平台的效率和易用性,加速算法同学的模型迭代效率。

更新于 2023-12-26
logo of nio
社招5-7年算法

1、参与LLM/语音/图像相关的大模型在数字座舱场景下的研发,包括模型的设计、训练和优化; 2、大模型相关上线工作,解决大模型业务落地的相关技术障碍,支撑各种落地应用包括但不限于openQA应用,相关QA应用,任务型pipeline分模块应用,端到端对话系统应用等。

更新于 2023-05-26