Jobs · OTHR · California

Research Scientist Graduate (AI Infra Compute) - 2027 Start (PhD)

ByteDance · San Jose, CA · 2 mo ago
OTHR$162k–$317k/yrFull-time

About the role

Compute division focuses on building large-scale and highly available cloud infrastructure, which supports both public cloud products (like VolcEngine ECS service) and internal products. The Compute-US team focuses on the development and research of the compute infrastructure platform. As a graduate, you will have opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth. Successful candidates must commit to an onboarding date by the end of the year.

Responsibilities

  • Develop key technologies to optimize our AI Infra stack, including training infra, inference infra, and AI agents.
  • Work with academia and open source communities on joint development.
  • Follow the latest technologies from academia or industry and conduct deep-dive analysis.
  • Present our research and products in academic papers.

Requirements

Minimum Qualifications:

  • Individuals who are completing or have recently completed a PhD degree in Computer Science, Computer Engineering, or a related discipline.
  • Experience with at least one of the following areas:
    • LLM training infra, including optimizations for various post-training workloads such as RL training, knowledge distillation, etc.
    • LLM inference infra, including inference engine performance improvements, more efficient execution parallelism, GPU kernel optimizations, etc.
    • AI Agent Infra, including computer-use agents, coding agents, agent memory, agent sandbox, etc.
  • Commit to proactive continuous learning, demonstrate enthusiasm for AI technologies, and exhibit a strong ability to quickly grasp and apply new technologies.
  • Good communication and teamwork skills.

Preferred Qualifications:

  • Frequent contributors or maintainers in AI Infra related open source communities such as vLLM or SGLang.
  • Having CS conference publications such as OSDI or MLSys.

Pay

The base salary range for this position is $162,000 - $316,800 annually. Compensation may vary depending on qualifications, skills, competencies, experience, and location. Base pay is one part of the total package, which may include additional discretionary bonuses/incentives and restricted stock units.

Benefits

  • Day one access to medical, dental, and vision insurance.
  • 401(k) savings plan with company match.
  • Paid parental leave.
  • Short-term and long-term disability coverage.
  • Life insurance.
  • Wellbeing benefits.
  • 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).

About Us

Founded in 2012, ByteDance's mission is to inspire creativity and enrich life. With a suite of products including TikTok, Lemon8, CapCut, and Pico, as well as platforms specific to the China market like Toutiao, Douyin, and Xigua, ByteDance makes it easier and more fun for people to connect with, consume, and create content.

ByteDance fosters an "Always Day 1" mindset, encouraging innovation and impact in a rapidly growing tech company. We value curiosity, humility, and a desire to make a difference.

Similar jobs