Machine Learning Backend Engineer Graduate (AML MLdev) - 2027 Start

ByteDance
Singapore
Full timeFunction: Data Science & Machine LearningSkills: ["Independent thinking","Problem decomposition","Performance troubleshooting","Practical optimization"]

Work on ByteDance’s Data AML machine learning mid-platform that powers recommendation/advertising training and inference for internal business units. Optimize computational performance by tuning inference/training bottlenecks, improving utilization, and building high-performance kernel libraries. Develop and enhance a mid-platform model compilation system using graph optimization, kernel fusion, scheduling, and code generation, partnering with business and algorithm teams on end-to-end performance diagnosis and solutions.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
ByteDance
ByteDance
3 days ago

Machine Learning Backend Engineer Graduate (AML MLdev) - 2027 Start

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 12 hours agoStatus: Live

Job Summary

Work on ByteDance’s Data AML machine learning mid-platform that powers recommendation/advertising training and inference for internal business units. Optimize computational performance by tuning inference/training bottlenecks, improving utilization, and building high-performance kernel libraries. Develop and enhance a mid-platform model compilation system using graph optimization, kernel fusion, scheduling, and code generation, partnering with business and algorithm teams on end-to-end performance diagnosis and solutions.
Location: Singapore
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Graduate level

Key Responsibilities

  • •Optimize computational performance for recommendation mid-platform models, including tuning inference/training bottlenecks and improving computing utilization.
  • •Design and develop high-performance kernel libraries with general-purpose and business-customized kernels.
  • •Build and improve the mid-platform model compilation system using techniques like graph optimization, kernel fusion, computation scheduling, and code generation.
  • •Perform full-stack performance analysis and bottleneck diagnosis with business and algorithm teams.
  • •Consolidate general-purpose performance optimization components, toolchains, and platform capabilities to enable internal business units.

Key Requirements

  • •Completing or recently completed a Bachelor’s or Master’s degree in computing or a related discipline.
  • •Familiar with mainstream model compilation stacks such as TVM, MLIR, and XLA, with experience developing and optimizing systems.
  • •Proficient in C/C++ development, including familiarity with assembly, CPU/GPU architecture, and cache mechanisms for high-performance kernel development.
  • •Understand deep learning framework execution (e.g., TensorFlow, PyTorch, OneFlow), including computational graph structure and inference/training execution, with experience implementing graph and compilation optimizations.
  • •Strong independent thinking and problem decomposition skills to diagnose and practically resolve complex performance bottlenecks.
Education:
Skills:Independent thinkingProblem decompositionPerformance troubleshootingPractical optimization
Tech Stack:CC++AssemblyCPUGPUCache mechanismTVMMLIRXLATensorFlowPyTorchOneFlowMxNet

Company Brief

ByteDance
Develops consumer internet and content platforms, including TikTok and other apps for short-form video, news, and entertainment. It also builds advertising, commerce, and creator tools that connect audiences, brands, and publishers across global markets.
Industry: Digital Media
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Established Company
Headquarters: Beijing, China
Founded: 2012
WebsiteLinkedIn