Student Researcher - Compiler (Seed Infra) - 2026 Start

ByteDance
San Jose
Workplace: OnsiteInternshipFunction: Research & Scientific (R&D)Education: bachelorsSkills: ["Collaboration","Performance analysis"]

Work with the Seed Infrastructures team to optimize AI compiler passes and accelerate training and inference across heterogeneous hardware. Develop MLIR-based compiler transformations for graph lowering, optimization, and code generation, and improve model execution on GPU and NPU accelerators. Support deployment pipelines from compilation to runtime integration, assist with distributed execution and communication optimization, and benchmark large-scale models across hardware backends.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
ByteDance
ByteDance
7 hours ago

Student Researcher - Compiler (Seed Infra) - 2026 Start

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI
Last checked: 1 day agoStatus: Live

Job Summary

Work with the Seed Infrastructures team to optimize AI compiler passes and accelerate training and inference across heterogeneous hardware. Develop MLIR-based compiler transformations for graph lowering, optimization, and code generation, and improve model execution on GPU and NPU accelerators. Support deployment pipelines from compilation to runtime integration, assist with distributed execution and communication optimization, and benchmark large-scale models across hardware backends.
Location: San Jose
Workplace: Onsite
Employment Type: Internship
Job Function: Research & Scientific (R&D)
Seniority: Intern level

Key Responsibilities

  • •Contribute to AI compiler optimizations for training and inference workloads.
  • •Develop and extend MLIR-based compiler passes for graph lowering, optimization, and code generation.
  • •Optimize model execution on GPU and NPU accelerators with a focus on performance, memory efficiency, and scalability.
  • •Support model deployment pipelines, including compilation, packaging, and runtime integration.
  • •Benchmark, profile, and analyze performance of large-scale models across different hardware backends.

Key Requirements

  • •Currently pursuing a Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or a related technical field.
  • •Experience using or developing open source LLM inference frameworks such as vLLM or SGLang, with proficiency in at least one deep learning framework (PyTorch, Megatron, DeepSpeed, or JAX).
  • •Understanding of modern computing systems (hardware, storage, and networking) and how they affect ML workloads.
  • •Familiarity with compilers or model optimization pipelines (e.g., PyTorch Dynamo) or related model execution workflows.
  • •Able to commit to working for 12 weeks in 2026.
Education:Bachelor's in Computer Science, Electrical Engineering
Skills:CollaborationPerformance analysis
Tech Stack:MLIRVLLMSGLangPyTorchMegatronDeepSpeedJAXPyTorch DynamoTorch.fxXLACUDATritonNCCLRDMAFSDPGSPMDGPUNPUTPUCUDA programming

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