返回岗位列表

壁仞科技

搜广推架构工程师

地点:上海 薪资:30-50K 日期:2026-06-26

岗位摘要

参与推理引擎、深度学习框架、分布式训练、深度学习算法等相关方向的技术调研、设计及原型开发;探索业界领先技术,如千卡、万卡集群上大模型(LLM、文生图、文生视频等)的分布式训练和推理,并在壁仞GPGPU上进行原型开发验证

岗位职责

  • 参与推理引擎、深度学习框架、分布式训练、深度学习算法等相关方向的技术调研、设计及原型开发
  • 探索业界领先技术,如千卡、万卡集群上大模型(LLM、文生图、文生视频等)的分布式训练和推理,并在壁仞GPGPU上进行原型开发验证
  • 针对搜索、广告、推荐等场景,研究大规模稀疏模型训练框架和预估引擎
  • 研究大模型驱动的新型搜广推算法和AI工程架构,并在壁仞GPGPU上进行原型开发验证

任职要求

  • 计算机、电子、数学及相关专业
  • 熟练使用C++/Python编程
  • 熟悉CUDA等异构编程、GPGPU架构者优先
  • 具备PyTorch、TensorFlow、PaddlePaddle、Megatron-LM、DeepSpeed、vLLM、SGlang等框架研发和优化经验者优先
  • 熟悉Wide&Deep、DLRM、DIEN、MMOE等传统广告/推荐场景算法者优先
  • 熟悉TIGER、GR、OneRec等大模型驱动的新型算法者优先
  • 熟悉大规模稀疏模型训练框架如AIBox、HugeCTR、TorchRec、DeepRec等者优先
  • 熟悉稀疏模型预估引擎架构设计和业务落地者优先

福利待遇

  • 薪资范围:30-50K·15薪
  • 工作地点:上海闵行区临港浦江科技广场16号楼
  • 经验不限,本科及以上学历

工作地址

上海闵行区临港浦江科技广场16号楼