Inactive
T

TikTok

Short-form video platform with ads

Efficient ML Engineer Research Intern - AI Platform

Applied Machine LearningPosted on Apr 11
Internship · Summer 2026San Jose, CA, USAIn-person

About the Job

Requirements

  • Currently pursuing a PhD in Computer Science, engineering or quantitative field
  • Proficient in C++/Python and high-performance coding
  • Expertise in diffusion models (Stable Diffusion/DiT) with deep understanding of computational bottlenecks and optimization methodologies
  • Proven experience in ≥1 domain: model compression (quantization/knowledge distillation), efficient architectures (MoE/sparse attention), generative alignment (RLHF/DPO)
  • Excellent communication and teamwork skills, capable of thriving in a fast-paced work environment

Responsibilities

  • Develop algorithm acceleration technologies for text-to-image/text-to-video models through knowledge distillation, model architecture redesign (dynamic MoE routing/sparse attention), and parameter-efficient design (low-bit quantization) to achieve order-of-magnitude efficiency gains
  • Lead generative model innovation with focus on diffusion acceleration (sampling step reduction/latent optimization), autoregression model efficiency
  • Collaborate cross-functionally to identify performance bottlenecks, optimize vision models via algorithmic breakthroughs, and enhance ByteDance's product capabilities

Desired Qualifications

  • Kaggle competition achievements, publications at ICML/NeurIPS/CVPR, or open-source contributions (e.g., HuggingFace Diffusers optimization)
  • Research experience in GenAI /MLsys areas
  • Familiarity with open source deep learning frameworks such as Pytorch/DeepSpeed/Jax etc

About the Company

TikTok

Short-form video platform with ads

TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.

Company size
10,001+
Company stage
Grant
Total funding
$740K
Headquarters
Santa Monica, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at TikTok

See more jobs