← 返回岗位列表美国fulltime

研究工程师 - [Seed模型 - 基础设施 - 训练性能与机器学习编译]

雇主

ByteDance

地点

California · San Jose · 美国

待遇

$面议

工作模式

线下

截止日期

1月8日

🤖 AI 简历匹配评估

检测你的简历与该岗位的匹配度，免费

免费评估

岗位摘要

Optimize large-scale foundation model training via ML compilation and GPU kernel development.

岗位职责

The Seed Infrastructures team oversees the distributed training, reinforcement learning framework, high-performance inference, and heterogeneous hardware compilation technologies for AI foundation models.
Responsibilities
- Optimize training performance for large-scale foundation models through compiler-level techniques, including graph optimization, operator fusion, and kernel generation.
- Develop and extend ML compilation capabilities based on the PyTorch compilation stack (e.g. FX, Dynamo, Inductor) to improve training efficiency across heterogeneous GPU platforms.
- Design and optimize high-performance GPU kernels for training workloads.
- Conduct performance profiling and analysis of large-scale training jobs; identify and resolve bottlenecks in collaboration with research and infrastructure teams.

申请条件

Minimum Qualification(s)
- Bachelor's degree or above in Computer Science, Electrical Engineering, or a related field.
- Strong proficiency in C/C++ and Python; solid foundations in algorithms, data structures, and systems programming.
- Hands-on experience in training-side performance optimization for deep learning workloads.
- Hands-on experience writing and optimizing GPU kernels (e.g., CUDA, Triton).
- Experience with the PyTorch compilation stack, meeting at least one of the following: Direct experience using, debugging, or extending Inductor or FX.
- Proficiency in Triton kernel development.
- Solid experience with PyTorch computation graph work (graph optimization, graph capture, operator fusion).
Preferred Qualification(s)
- Experience with TorchDynamo or bytecode-level program transformation.
- Experience with Triton compiler internals or other ML compiler backends (e.g., MLIR, LLVM).
- Contributions to related open-source projects (e.g., PyTorch, Triton, FlashAttention).
- Publications in relevant venues (e.g., MLSys, OSDI, ASPLOS).

雇主简介

ByteDance is a global technology company known for its content platforms, and its infrastructure team builds and manages hyper-scale data centers worldwide, focusing on power and energy management for data center operations.

对这个岗位感兴趣？

该岗位通过项目服务流程申请，顾问可为您介绍项目详情与申请条件

咨询不收取任何费用，项目顾问将与您联系

申请海外岗位，英文简历符合当地格式规范吗？

AI 自动评估你与该岗位的匹配度，3 分钟出结果

免费评估简历匹配度

数据来源：ByteDance

岗位信息来源于公开渠道，版权归原作者所有

← 返回岗位列表