字节跳动
校招AI异构计算优化工程师 - Data AML
杭州/上海/北京 · 校招 · 2027 届 · 本科
长期有效
任职要求
- 2027届获得本科及以上学历
- 熟练掌握Linux环境下的C/C++、Python语言；能够熟练使用至少一种主流的机器学习框架（TensorFlow/PyTorch等），熟悉框架内部实现
- 具备扎实的计算机科学功底和编程能力，熟悉常见算法和数据结构，具有良好的编程习惯；熟悉至少一种经典深度学习模型及其应用场景，如GPT、SD、DiT、W&D等
- 有良好的工作文档习惯，及时按要求撰写更新工作流程及技术文档；了解推理/训练/通信相关优化，有高性能算子、大规模训练、通算融合领域经验者优先
- 了解SIMD/SIMT外其他并行计算模型及算法，了解其并行编程模型及其优、缺点者优先；了解模型剪枝、量化、LLM投机采样等优化方法原理，具备相关模型优化经验者优先
- 了解常见AI编译器方案及其优、缺点（如XLA、TVM、MLIR），有实际开发经验者优先；了解GPU硬件架构，了解GPU软件栈（CUDA、cuBLAS、CUTLASS），具备GPU性能分析的能力者优先；了解AI Compiler技术栈（MLIR等），熟悉PyTorch
- 0+编译栈、Triton者优先。
岗位描述
团队介绍：Data AML是字节跳动的机器学习中台，为抖音/今日头条/西瓜视频等业务提供推荐/广告/CV/语音/NLP的训练和推理系统。为公司内业务部门提供强大的机器学习算力，并在这些业务的问题上研究一些具有通用性和创新性的算法。同时，也通过火山引擎将一些机器学习/推荐系统的核心能力提供给外部企业客户。
- 1、负责评估符合业务要求的异构计算芯片，建立并完善评估体系与负载模型，预估业务收益；
- 2、负责异构计算芯片在实际推理业务中的落地，适配芯片特性以降低推理时延、提高推理吞吐；
- 3、负责异构计算芯片在实际训练业务中的落地，结合芯片计算与通信特性优化显存占用、提升训练吞吐；
- 4、负责开发异构计算芯片的高性能算子，依据芯片特性优化算。
你的简历匹配这个岗位吗？
上传简历，AI 会对照这份 JD 指出你的差距和需要补强的地方
该岗位可能会问的问题
基于 算法 岗位的常见面试问题整理
请解释 Transformer 模型的自注意力机制（Self-Attention）
查看答题思路
1. Self-Attention = 序列中每个 token 与所有 token 计算相关性权重 2. Q=Wq·x, K=Wk·x, V=Wv·x; Attention(Q,K,V)=softmax(QK^T/√d_k)V 3. 除以 √d_k 防止点积过大导致 softmax 梯度消失 4. 多头注意力：多组 QKV 并行，拼接后线性变换，捕获不同子空间特征 5. 对比 RNN：可并行计算，捕获长距离依赖
你做过的最有深度的算法/模型项目是什么
查看答题思路
1. 一句话说清任务：分类/生成/检索/推荐 2. 模型选型：为什么选这个而非那个 3. 数据：数据量、来源、清洗策略 4. 创新点：你做的改进是什么（不是调参） 5. 结果：核心指标提升（准确率/召回/延迟）
过拟合是什么？怎么判断和解决
查看答题思路
1. 过拟合=训练集表现好、验证集表现差，模型记住了噪声 2. 判断：train loss 持续下降而 val loss 上升 3. 解决：更多数据 / 正则化(L1/L2) / Dropout / 早停 / 数据增强 / 降低模型复杂度
解释梯度下降的原理，SGD 和 Adam 的区别
查看答题思路
1. 梯度下降：沿负梯度方向更新参数，使损失降低 2. SGD：每次用随机小批量估算梯度，引入随机性帮助跳出局部最优 3. Adam = Momentum + RMSProp：自适应学习率 + 动量加速 4. Adam 收敛快但可能泛化不如 SGD；CV 常用 SGD，NLP 多用 Adam
给你 10 万条未标注的文本数据，如何找出其中 10 个主题
查看答题思路
1. 预处理：分词、去停用词 2. 向量化：TF-IDF 或 Sentence Embedding 3. 聚类：K-Means（需指定 K=10）或 HDBSCAN（自动确定类数） 4. 或 LDA 主题模型直接输出主题-词分布 5. 评估：看每个簇的高频词判断主题合理性，迭代调整
更多面试题库功能即将上线
信息来源：企业官方招聘页。投递前请以官网信息为准。