Senior GenAI Research Engineer: Distillation & Efficiency

TikTok

Seattle (WA)

On-site

USD 242,000 - 456,000

Full time

9 hours ago

Be an early applicant

Application generator

A complete application in a minute — tailored resume and cover letter, ready to send.

Get past ATS filters

Benefits offered by this job

Health insurance

Dental insurance

Vision insurance

401(k) with company match

Paid parental leave

Disability insurance

Life insurance

Wellbeing benefits

10 holidays per year, 10 sick days, 17

Paid Personal Time

Job summary

TikTok's Vision-Applied Research team seeks a Research Engineer/Scientist to design efficient models for large-scale generative AI, focusing on distillation and compression. You will develop methods and infrastructure to transfer capabilities from foundation models into smaller, deployment-friendly variants for scalable training and inference.

Responsibilities include distillation frameworks, model acceleration, and hardware-efficient inference, with emphasis on diffusion and autoregressive

Qualifications

BS in CS or related field or equivalent experience
Expertise in efficient models with understanding bottlenecks and acceleration
Experience training generative AI/LLM models using PyTorch and JAX
Strong communication and collaboration in fast-paced environments

Responsibilities

Design and implement efficient models for large-scale generative AI, focusing on distillation and compression
Develop methods and infrastructure to transfer capabilities from foundation models into smaller models for scalable training and deployment
Work on distillation frameworks, model acceleration, hardware-efficient inference and applications