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

ByteDance; San Jose (CA); On-site; USD 180,000 - 230,000; Full time

Job summary

ByteDance Technology in San Jose seeks PhD-level researchers for a Compute division role. You will help build scalable AI infra, spanning training and inference, and contribute to AI agents. This position emphasizes explorative research, collaboration with universities, and publishing results in academic venues.

As part of ByteDance, you will join a team that values rapid learning, rigorous analysis, and cross-functional collaboration to advance cloud infrastructure for global products.

Qualifications

• PhD in Computer Science, Computer Engineering, or related discipline.
• Experience with LLM training infra (RL training, distillation, etc.).
• Experience with LLM inference infra (inference engine performance, GPU kernel optimizations).
• Experience with AI Agent Infra (agent memory, sandbox, etc.).
• Commit to proactive continuous learning and ability to quickly grasp new technologies.
• Good communication and teamwork skills.
• Preferred: frequent contributors or maintainers in AI Infra open source (e.g., vLLM, SGLang).
• Preferred: CS conference publications such as OSDI or MLSys.

Responsibilities

• Develop key technologies to optimize AI Infra stack for training, inference, and AI agents.
• Collaborate with academia and open source communities on joint development.
• Follow latest research and conduct deep-dive analyses.
• Present research and products in academic papers.

Skills

LLM training infra

LLM inference infra

AI Agent Infra

Proactive learning

Communication

Teamwork

Open source collaboration

Academic publication

Education

PhD in Computer Science or related field

Job description

Join us as we work together to inspire creativity and enrich life around the globe.

Location:

San Jose

Team:

Technology

Employment Type:

Regular

Job Code:

A16363A

Share this listing:

Responsibilities

About the teamCompute division focuses on building large-scale and highly available cloud infrastructure, which supports both public cloud products (like VolcEngine ECS service) and the internal products. The Compute-US team focuses on the development and research of the compute infrastructure platform.We are looking for talented individuals to join our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation date clearly in your resume.

• 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.

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
• Experience with at least one of the following areas: LLM inference infra, including inference engine performance improvements, more efficient execution parallelism, GPU kernel optimizations, etc
• Experience with at least one of the following areas: 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.
• Preferred Qualifications: Having CS conference publications such as OSDI or MLSys.

Job Information

About Us

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

Why Join ByteDance

Inspiring creativity is at the core of ByteDance's mission. Our innovative products are built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and enrich life - a mission we work towards every day.

As ByteDancers, we strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our Company, and our users. When we create and grow together, the possibilities are limitless. Join us.

Diversity & Inclusion

ByteDance is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At ByteDance, our mission is to inspire creativity and enrich life. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

Reasonable Accommodation

ByteDance is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at https://tinyurl.com/RA-request