San Jose, CA

Large Language Model Inference System Engineer Graduate (Applied Machine Learning) - 2027 Start

Onsite$128,000 - $256,000/yrPosted 2 weeks agoVisa Sponsorship

Job details

Location
San Jose, CA
Work type
Onsite
Compensation
$128,000 - $256,000/yr
Visa
Sponsorship available
Posted
2 weeks ago
Apply on
joinbytedance.com

About this role

ByteDance is hiring a graduate engineer to develop and optimize the Volcano Ark MaaS inference system. The role focuses on improving large-model inference performance, cost, and stability across large-scale heterogeneous clusters.

What you'll do:

  • Participate in the engineering development of the Volcano Ark MaaS inference system, optimizing large model inference performance, cost, and stability in ultra-large-scale heterogeneous inference clusters
  • Reduce large model inference costs through system-level approaches such as disaggregated multi-role inference, distributed KV Cache systems, heterogeneous inference, elastic computing, and multi-tenant co-located inference

What they're looking for:

  • Individuals who are completing or have recently completed a Bachelor's or Master's degree in Computer Science or a related discipline
  • Strong command of algorithms, design patterns, and data structures, with solid knowledge of operating systems and computer architecture
  • Proficient in one or more programming languages such as C++ or Python, with good coding style
  • Understands GPU hardware architecture, is familiar with high-performance computing software stacks such as CUDA, and has experience in GPU performance analysis
  • Strong interest in distributed systems and large-scale heterogeneous inference; enjoys studying low-level principles and performance bottlenecks, and actively follows progress in related fields
  • Experience in large model inference system optimization, with practical understanding of PD disaggregation, KV Cache systems, and multi-node inference
  • Experience in distributed network communication optimization, with deep understanding of distributed communication operator implementation and RDMA principles
  • Familiarity with resource orchestration and scheduling frameworks such as Kubernetes and Ray

Benefits:

  • May be eligible for additional discretionary bonuses/incentives and restricted stock units.
  • Employees have day one access to medical, dental, and vision insurance, a 401(k) savings plan with company match, paid parental leave, short-term and long-term disability coverage, life insurance, and wellbeing benefits. Benefits may vary depending on the nature of employment and the country work location.
  • 10 paid holidays per year.
  • 10 paid sick days per year.
  • 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).
Ready to apply to ByteDance?
ApplyBolt finds matching jobs, tailors your resume, and submits applications for you.

About ByteDance