Student Researcher (AI Foundation Models Infrastructure - Seed Infra) - 2026 Start (PhD)

ByteDance
Seattle
InternshipFunction: Research & Scientific (R&D)Education: phdSkills: []

Join the Seed Infrastructures team to help advance distributed training and post-training systems for AI foundation models. You may design and optimize large-scale distributed training, contribute to reinforcement learning training frameworks, improve inference latency and throughput, and develop compiler/runtime optimizations for heterogeneous GPU/accelerator hardware. Work on system performance analysis, profiling, bottleneck diagnosis, and build tooling to boost developer productivity and system reliability.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
ByteDance
ByteDance
2 hours ago

Student Researcher (AI Foundation Models Infrastructure - Seed Infra) - 2026 Start (PhD)

✓ Verified Job

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

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

Job Summary

Join the Seed Infrastructures team to help advance distributed training and post-training systems for AI foundation models. You may design and optimize large-scale distributed training, contribute to reinforcement learning training frameworks, improve inference latency and throughput, and develop compiler/runtime optimizations for heterogeneous GPU/accelerator hardware. Work on system performance analysis, profiling, bottleneck diagnosis, and build tooling to boost developer productivity and system reliability.
Location: Seattle
Employment Type: Internship
Job Function: Research & Scientific (R&D)
Seniority: Intern level

Key Responsibilities

  • •Design and optimize large-scale distributed training systems, including parallelism, memory efficiency, and fault tolerance.
  • •Contribute to reinforcement learning training frameworks and large-scale post-training systems.
  • •Improve inference performance, latency, and throughput for foundation models.
  • •Develop compiler or runtime optimizations for heterogeneous hardware (GPU/accelerator).
  • •Perform system-level performance analysis, profiling, bottleneck diagnosis, and build tooling/automation for reliability and developer productivity.

Key Requirements

  • •Currently pursuing a PhD in Computer Science, Electrical Engineering, or a related technical field.
  • •Strong programming skills in Python and/or C++.
  • •Solid understanding of systems, distributed computing, machine learning systems, or performance optimization.
  • •Experience with distributed training frameworks such as PyTorch FSDP or Megatron-style parallelism.
  • •Experience with reinforcement learning training systems and/or GPU programming (CUDA, Triton) or compiler technologies.
Experience:Machine learning systemsDistributed computingAI infrastructureOpen-sourceGPU programming
Education:PhD / Doctorate in Computer Science, Electrical Engineering, or related technical fields
Tech Stack:PythonC++PyTorch FSDPMegatron-style parallelismCUDATritonGPU programmingCompiler technologiesDistributed training frameworks

Eligibility

Work Authorization:Authorization required. Sponsorship not provided.

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