Efficient ML Engineer Research Intern, (AI Platform) - 2026 Start (PhD)

TikTok is the leading destination for short-form mobile video, and they are seeking a PhD intern to contribute to their AI platform. The role involves developing algorithm acceleration technologies for GenAI models and collaborating cross-functionally to enhance product capabilities.

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