深度学习算法测试专家
岗位摘要
搭建并优化深度学习大模型的测试体系,确保模型的可扩展性和推理性能;开发和优化基于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楼
登录后查看完整岗位详情
使用邮箱验证码登录后,可查看完整职责、任职要求、原岗位入口,并继续使用收藏和浏览记录。列表摘要保持公开,完整详情用于保护数据质量和降低恶意抓取。