返回岗位列表

寒武纪

深度学习算法测试专家

地点:北京 薪资:25-50K 日期:2026-03-03

岗位摘要

搭建并优化深度学习大模型的测试体系,确保模型的可扩展性和推理性能;开发和优化基于TensorFlow、PyTorch、DeepSpeed等框架的大模型测试平台;推动分布式训练、混合精度训练、推理加速等技术的落地

岗位职责

  • 搭建并优化深度学习大模型的测试体系,确保模型的可扩展性和推理性能
  • 开发和优化基于TensorFlow、PyTorch、DeepSpeed等框架的大模型测试平台
  • 推动分布式训练、混合精度训练、推理加速等技术的落地
  • 深入理解大模型推理技术难点,提出优化方案并协助解决模型压缩、推理延迟、吞吐量瓶颈等问题
  • 基于大模型特性(如模型剪枝、量化、蒸馏等),从测试角度提供针对性优化建议
  • 通过AutoML、模型压缩、量化训练、稀疏性优化等技术手段解决复杂技术难题

任职要求

  • 精通C/C++/Python,具备良好的编程风格与代码规范,能够独立完成大模型测试工具和平台的开发
  • 熟悉大模型(如GPT、BERT、T5等)、Transformer架构,了解知识蒸馏、模型量化、模型压缩等优化技术
  • 有大规模深度学习模型的推理优化经验,精通分布式训练(如Horovod、DeepSpeed)、GPU/TPU加速等技术
  • 熟悉PyTorch、TensorFlow等主流框架,具备大模型推理、推理加速、混合精度训练等技术实践经验
  • 具备清晰的逻辑思维能力,能够快速识别技术问题并提出有效的解决方案
  • 熟悉LLVM开发或参与自定义编程语言开发者优先
  • 有AI芯片(如NPU、TPU)测试经验者优先考虑
  • 参与过大规模并行计算、分布式系统优化、高性能计算等相关项目者优先

加分项

  • 熟悉 LLVM 开发或参与自定义编程语言开发者优先
  • 有 芯片相关测试经验,特别是 AI芯片(如 NPU、TPU)的测试经验者优先考虑
  • 参与过 大规模并行计算、分布式系统优化、高性能计算 等相关项目者优先
  • 我们希望你: 热衷于 大模型 和 深度学习领域,并且渴望在 推理加速、分布式训练 等前沿技术中不断提升自己的技术能力
  • 在复杂的技术挑战面前能够保持冷静,并通过系统化、工程化的思维进行问题解决
  • 有强烈的责任感和技术驱动力,能够主动推动技术革新,并为团队带来持续的技术积累
  • 我们能提供: 具有竞争力的薪资待遇和全面的职业发展机会
  • 与一流AI团队一起合作,参与 大模型、推理优化、分布式训练 等前沿技术研发
  • 开放和包容的工作环境,注重员工成长与技术交流
  • 如何加入我们: 如果你热衷于深度学习、AI大模型和高效能计算,具备扎实的技术背景,并且希望在具有挑战性的工作环境中发挥技术领导力,欢迎投递简历,和我们一起突破技术边界,开创AI未来!
  • 李先生 刚刚活跃
  • 寒武纪 · 软件研发部门经理

福利待遇

  • 具有竞争力的薪资待遇和全面的职业发展机会
  • 与一流AI团队一起合作,参与大模型、推理优化、分布式训练等前沿技术研发
  • 开放和包容的工作环境,注重员工成长与技术交流

工作地址

北京海淀区北航致真大厦D座11楼