ByteDance

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

ByteDance
Apply
2 hours ago
San Jose, CA, USAEntry Level
H1B sponsor

Responsibilities

  • Develop technologies that optimize AI infrastructure, including training infrastructure, inference infrastructure, and AI agent infrastructure.
  • Conduct optimization work involving post-training workloads, inference engine performance, execution parallelism, and GPU kernels.
  • Collaborate with academic institutions and open-source communities on joint development.
  • Analyze emerging technologies from academia and industry.
  • Present research and products in academic papers.

Requirements

  • Be completing or have recently completed a PhD in Computer Science, Computer Engineering, or a related discipline.
  • Have experience with at least one of LLM training infrastructure, LLM inference infrastructure, or AI agent infrastructure.
  • Demonstrate continuous learning, enthusiasm for AI technologies, and the ability to quickly learn and apply new technologies.
  • Have good communication and teamwork skills.
  • Frequent contributions to or maintainership of AI infrastructure open-source communities such as vLLM or SGLang are preferred.
  • Publications at computer science conferences such as OSDI or MLSys are preferred.
  • State availability and graduation date clearly in the resume and commit to an onboarding date by the end of the year.
ByteDance

About ByteDance

10,000+ employees

ByteDance is a global incubator of platforms at the cutting edge of commerce, content, entertainment and enterprise services - over 2.5bn people interact with ByteDance products including TikTok. Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This is doubly true of the teams that make our innovations possible. Together, we inspire creativity and enrich life - a mission we aim towards achieving every day. At ByteDance, we create together and grow together. That's how we drive impact - for ourselves, our company, and the users we serve. We are committed to building a safe, healthy and positive online environment for all our users. We have over 110,000 employees based in more than 30 countries globally. Join us.

Contact me