B

ByteDance

Builds content apps, business software and AI

Research Intern - AI Compute

Platform EngineeringConfirmed live in the last 24 hours
Internship · Summer 2027San Jose, CA, USANo salary listedIn-person

About the Job

Requirements

  • Currently pursuing a PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related technical field.
  • Able to commit to working for 12 weeks during Summer 2027.
  • Strong understanding of large model inference, distributed and parallel systems, and/or high-performance networking systems.
  • Hands-on experience building cloud or machine learning infrastructure in areas such as resource management, scheduling, request routing, monitoring, or orchestration.
  • Solid knowledge of container and orchestration technologies, including Docker and Kubernetes.
  • Proficiency in at least one major programming language: Go, Rust, Python, or C++.

Responsibilities

  • Design and build large-scale, container-based cluster management and orchestration systems with extreme performance, scalability, and resilience.
  • Architect next-generation cloud-native GPU and AI accelerator infrastructure to deliver cost-efficient and secure machine learning platforms.
  • Collaborate across teams to deliver inference solutions using vLLM, SGLang, TensorRT-LLM, and other large language model engines.
  • Stay current with advances in open source, including Kubernetes and Ray, AI/machine learning and large language model infrastructure, and systems research; integrate best practices into production systems.
  • Write high-quality, production-ready code that is maintainable, testable, and scalable.

Desired Qualifications

  • Experience contributing to or operating large-scale cluster management systems, such as Kubernetes or Ray.
  • Experience with workload scheduling, GPU orchestration, scaling, and isolation in production environments.
  • Hands-on experience with GPU programming using CUDA or inference engines such as vLLM, SGLang, or TensorRT-LLM.
  • Familiarity with public cloud providers such as AWS, Azure, or GCP and their machine learning platforms, including SageMaker, Azure ML, or Vertex AI.
  • Strong knowledge of machine learning systems, including Ray, DeepSpeed, and PyTorch, and distributed training or inference platforms.
  • Excellent communication skills and ability to collaborate across global, cross-functional teams.
  • Passion for system efficiency, performance optimization, and open-source innovation.

About the Company

ByteDance

Builds content apps, business software and AI

ByteDance builds content and creation apps for consumers, alongside collaboration software and AI services for businesses. Its video platforms connect viewers with creators, while CapCut gives people tools to edit and publish videos. The group also develops Lark for workplace communication and shared documents, and its Seed research teams build language, image, video, and speech models used in consumer apps and sold through BytePlus cloud services. It earns money from advertising, commerce services, paid software subscriptions, and usage of its enterprise technology. ByteDance was founded in 2012 and has its operational headquarters in Beijing, China.

Company size
10,001+
Company stage
Private
Total funding
$35.1B
Headquarters
Haidian, China
Founded
2012

Recently Posted Jobs

Explore more opportunities at ByteDance

See more jobs