Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Student Researcher - Compiler - 2026 Start

Home > Python programming jobs

Student Researcher - Compiler - 2026 Start in USA

  • ByteDance
  • Full time
  • Email
  • San Jose, CA

Responsibilities

  • Develop 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 for performance, memory efficiency, and scalability.
  • Support model deployment pipelines through compilation, packaging, and runtime integration.
  • Assist with distributed training and inference acceleration, including parallel execution, communication optimization, and runtime scheduling.
  • Benchmark, profile, and analyze large-scale models across different hardware backends.
  • Collaborate with researchers and engineers to translate model and system requirements into compiler and runtime improvements.

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.
  • Proficiency in at least one deep learning framework, such as PyTorch, Megatron, DeepSpeed, or JAX, with experience in model inference workflows.
  • Understanding of modern computing systems, including hardware, storage, and networking, and their impact on ML workloads.
  • Familiarity with compilers, model optimization pipelines such as PyTorch Dynamo, or related model execution workflows.
  • Able to commit to working for 12 weeks in 2026.
  • Preferred: experience with distributed or large-scale ML systems and training or inference optimizations such as FSDP, DeepSpeed, Megatron, or GSPMD.
  • Preferred: experience with GPU, TPU, or NPU programming and performance optimization, high-performance computing, or communication technologies such as CUDA, Triton, NCCL, or RDMA.
  • Preferred: understanding of AI compiler and model optimization stacks such as torch.fx, PyTorch Dynamo, XLA, and MLIR.

Benefits

  • 12-week work commitment during 2026.

ByteDance

ByteDance is a global incubator of platforms at the cutting edge of commerce, content, entertainment and enterprise services - over 2.5bn people interact with ByteDance products including TikTok. Creation is the core of ByteDance's purpose. Our products are built to help imaginations thrive. This...

Similar positions

Research Scientist - Machine Learning

  • ISEE
  • Full time
  • USA
  • 09/13/2026
  • Remote

Principal Software Engineer - Builder Experience

  • Elastic
  • Full time
  • Remote
  • 09/13/2026
  • Salary: €73k-€116k
  • Greece

Principal Software Engineer - Builder Experience

  • Elastic
  • Full time
  • Remote
  • 09/13/2026
  • Salary: zł 369k-zł 584k
  • Poland

Platform Engineer (Full-Stack Generalist)

  • DXC Technology
  • Full time
  • Others
  • 09/13/2026
  • Ariana, Tunisia

Staff Backend Engineer - Submission Processing

  • Kalepa
  • Full time
  • Remote
  • 09/13/2026
  • Salary: $100k-$185k
  • Europe

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026