ByteDance
Machine Learning Backend Engineer Graduate (AML MLDev) - 2027 Start
San Jose, California · Full-time · Posted Oct 28, 2026
- Education
- Completing or recently completed a Bachelor's/Master's degree in Software Development, Computer Science, Computer Engineering, or a related technical discipline.
About the job
Optimize performance for ByteDance’s recommendation machine-learning platform by developing high-performance kernels, improving model compilation, and diagnosing training and inference bottlenecks.
What you'll do
- Tune training and inference performance and improve computing utilization.
- Design and develop general-purpose and business-customized kernel libraries.
- Improve the model compilation system and provide performance analysis, bottleneck diagnosis, and optimization tools for internal teams.
What they ask for
- Completing or recently completed a Bachelor's or Master's degree in Software Development, Computer Science, Computer Engineering, or a related technical discipline.
- Development and optimization experience with model compilation stacks such as TVM, MLIR, or XLA.
- Proficiency in C/C++; familiarity with assembly, CPU/GPU architecture, and cache mechanisms; practical high-performance kernel development experience.
- Knowledge of deep-learning framework internals and experience implementing model graph and compilation optimizations.
Nice to have
- Experience in joint hardware/software design or heterogeneous computing projects.
- Contributions to or development of open-source deep-learning kernel libraries, compilers, or inference engines; or in-depth research on a machine-learning framework’s architecture.
Tools
TVM, MLIR, XLA, C/C++, TensorFlow, PyTorch, OneFlow, MxNet
AI focus
Develop and optimize machine-learning training and inference systems, kernels, and model compilation for recommendation workloads.
Model-written summary · the original posting is authoritative