硬件加速推理AI Infra工程师（Data） - 芯片研发

关于这个机会

团队介绍：字节芯片研发团队目前工作主要集中在芯片设计环节，主要围绕字节自身业务展开芯片探索，为字节多项业务的专用场景定制硬件优化，设计多款基于先进半导体工艺的云端复杂芯片，以期提升性能、降低成本。早期若干芯片项目已经进入到量产部署阶段，有多次一版成功的投片经历，所用工艺包含多个主流的先进工艺节点。与基础设施各团队之间工作的整体协同，能更容易和更好地发挥芯片研发的价值。 1、负责大模型等业务模型在自研芯片进行适配性和性能评估、软硬件协同优化工作； 2、负责AI模型整图或者算子编译框架优化，实现较高硬件利用率和整体性能，提升编译框架的通用性与优化能力； 3、构建分布式推理系统，推理框架极致调优，实现较优的在线部署吞吐和高性价比； 4、模型量化、稀疏、蒸馏等推理部署的探索和落地方案。

任职要求

1、深入理解AI加速硬件的体系结构和并行计算； 2、熟悉深度学习框架和软件栈，熟练使用C/C++、Python等编程语言； 3、熟悉ONNX/TensorFlow/PyTorch等主流深度学习框架； 4、有主动学习、问题分析与解决能力和较强自驱力；具备良好的跨团队协作能力。 以下为加分项： 1、熟悉编译器和代码优化技术，有MLIR、TVM等相关开发经验； 2、熟悉GPU或者主流AI芯片架构，有相关算子优化、图编译优化经验、硬件评测经验； 3、对LLM或多模态领域模型有深刻理解，例如Qwen、Deepseek等； 4、有AI加速硬件的量化工具开发经验； 5、熟悉AI服务器和集群架构，熟悉芯片和集群互联相关技术，有实践开发经验； 6、有vLLM/SGLang等推理框架开发和调优经验，有部署实践相关经验。

内推申请说明

本站为独立内推协助平台。申请会交由管理员核实岗位与内推渠道，不等于已在公司官网投递；薪资、岗位状态和实际招聘流程以官方信息为准。