招聘中 校招

大模型训推性能优化工程师 - AI算力基础设施

字节跳动

  • 北京、杭州、西安
  • 后端

收录时间

部门介绍

字节跳动基础设施部门,负责字节跳动的全栈算力基础设施,从芯片服务器产研、到超大规模数据中心,到传统云平台和如今的AI Native Cloud,全方位为抖音、豆包、今日头条、飞书、火山引擎等各类产品提供领先、稳定的百万量级大规模算力基础设施服务。我们的领域涵盖数据中心建设、内核操作系统开发、服务器集群管理、高速网络通信、EB级数据存储体系、搜索型数据库探索、基于LLM的AI基础设施的研发调度与治理等,致力于打造业界领先的、面向LLM的AI云原生基础设施架构与产品矩阵。

  1. 参与字节跳动大模型训练与推理全链路性能优化,聚焦千亿/万亿参数规模模型的训练吞吐、推理时延与资源效率提升;参与模型优化方向,LLM与多模态模型推理优化,包括混合精度量化(INT
  2. INT
  3. FP8)、稀疏化、Sparse Attention等稀疏注意力机制优化技术,平衡推理性能、模型精度与资源开销;参与主流训推框架(PyTorch、vLLM、SGLang等)与自研框架的适配、优化与二次开发,落地投机采样(Speculative Decoding)算法优化,优化专家路由策略与负载均衡机制,设计计算-通信精细化调度与重叠(Overlap)等方案,提升MOE架构模型推理吞吐量;
  4. 参与通信优化方向:AI框架跨节点/集群通信性能攻坚,优化RDMA/TCP协议栈传输机制、低延迟序列化方案与通信拓扑设计,突破分布式场景下的通信瓶颈;
  5. 参与计算优化方向:Transformer核心算子、卷积、矩阵乘加(GEMM)等AI核心算子优化,基于Triton/MLIR编译框架实现算子融合、向量化执行与编译优化;
  6. 参与GPU/NPU等异构算力平台的性能建模、性能剖析与代码级调优,适配GPU CUDA/ROCm及国产AI芯片专用指令集,最大化硬件算力利用率;
  7. 参与构建AI Infra性能Benchmark体系,形成量化性能标尺与竞争力评估模型。

岗位职责

招聘官网暂未提供完整岗位职责,请前往官网核实。

职位要求

  1. 2027届获得本科及以上学历,计算机、软件工程、电子信息、微电子等相关专业优先;
  2. 掌握扎实的计算机基础与算法功底,熟练掌握C/C++、Python、Go中的至少一种编程语言;
  3. 了解LLM/VLM/AIGC等AI领域基础原理与典型模型架构,理解训练与推理的基本流程;
  4. 有PyTorch等主流深度学习框架的使用经验,参与过模型训练、参数调优或框架组件的二次开发;
  5. 对异构计算、高性能网络或存储IO的性能分析具备理论基础与研究兴趣,愿意深入钻研性能优化工具的原理;
  6. 具备良好的沟通协作能力、学习能力与工程责任心。 加分项
  7. 具备CUDA/Triton等GPU编程经验,或熟悉TensorRT/CUTLASS等性能加速工具,理解其加速原理;
  8. 在学术论文、课程设计或竞赛中涉及AI系统性能分析相关工作,如训练瓶颈定位、通信优化、并行策略、算子融合、量化与精度调优等;
  9. 有开源社区贡献经历,参与过AI框架、推理引擎或性能优化工具的代码提交与维护;
  10. 接触过模型量化、精度校准、并行训练、通信优化、投机采样、MOE路由等相关方向;
  11. 关注国产AI加速卡技术生态,具备相关文献研究或实验环境下的应用测试经验;
  12. 熟悉主流AI加速硬件(NVIDIA/AMD GPU、国产AI芯片)的架构特性,或使用过PyTorch Profiler、NVIDIA Nsight等性能分析工具。

有些机会,只在官网短暂出现

真正值得关注的岗位,常常只在企业官网短暂开放,可能两三天后就下线,也未必会同步到综合招聘平台。没有持续关注,你甚至不会知道它曾经出现。职先机持续聚合并核验官网岗位,帮你抓住职场先机,快人一步。

微信小程序 / 当前岗位

在微信里继续看这个岗位

字节跳动大模型训推性能优化工程师 - AI算力基础设施

扫码直达当前岗位收藏、浏览记录和下线提醒留在微信里
电脑端可直接微信扫码;手机端可保存小程序码后在微信中识别。
微信扫码在职先机小程序查看大模型训推性能优化工程师 - AI算力基础设施正在准备岗位码
职先机微信小程序一岗一码 · 正式版直达保存小程序码

岗位提醒 · 01

收藏当前岗位,下线及时通知

当前岗位大模型训推性能优化工程师 - AI算力基础设施字节跳动 · 北京、杭州、西安

01

扫码进入当前岗位无需重新搜索,直接打开当前岗位详情。

02

收藏并开启提醒在小程序内收藏后,岗位下线或变化时及时通知。

03

继续查看相似机会原岗位变化时,可继续浏览相关在招岗位。

职先机会持续核验公开岗位;岗位状态与最终招聘结果仍以企业招聘官网为准。

微信小程序扫码打开当前岗位

扫码打开当前岗位,完成收藏并开启下线提醒。

微信扫码在职先机小程序收藏大模型训推性能优化工程师 - AI算力基础设施正在准备岗位码

微信小程序 · 当前岗位

请前往微信小程序提交内推申请

字节跳动大模型训推性能优化工程师 - AI算力基础设施校招 · 北京、杭州、西安

01进入当前岗位扫码或打开后直达本岗位内推申请页。

02提交 PDF 简历简历替换、处理进度和消息提醒在小程序内同步。

手机端将尝试直接打开微信;电脑端请使用微信扫码。

微信扫码进入大模型训推性能优化工程师 - AI算力基础设施内推申请页正在准备岗位码
当前岗位专属码扫码后无需重新搜索岗位保存小程序码

内推提示

暂不支持该岗位内推

如需投递,建议前往官网进行投递。