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

ByteDance
San Jose
Workplace: OnsiteInternshipFunction: Research & Scientific (R&D)Education: phdSkills: []

Work on infrastructure for AI foundation models, supporting distributed training and reinforcement learning workflows, improving inference performance, and diagnosing system bottlenecks. You’ll develop compiler/runtime optimizations for heterogeneous GPU/accelerator hardware and build tooling and automation to enhance developer productivity and system reliability, collaborating across large-scale ML and systems components.

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

Work on infrastructure for AI foundation models, supporting distributed training and reinforcement learning workflows, improving inference performance, and diagnosing system bottlenecks. You’ll develop compiler/runtime optimizations for heterogeneous GPU/accelerator hardware and build tooling and automation to enhance developer productivity and system reliability, collaborating across large-scale ML and systems components.
Location: San Jose
Workplace: Onsite
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 such as GPUs/accelerators.
  • •Perform system-level performance analysis, profiling, bottleneck diagnosis, and build reliability/productivity tooling.

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 systemsReinforcement learningML infrastructureOpen-source
Education:PhD / Doctorate
Tech Stack:PythonC++PyTorch FSDPMegatronCUDATriton

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