Senior GenAI Research Engineer—Efficient Large Models

TikTok; Seattle (WA); On-site; USD 242,000 - 456,000; Full time

Job summary

TikTok’s Vision-Applied Research team seeks a Research Engineer/Scientist to design and implement efficient large-scale generative AI models, emphasizing distillation and compression to move capabilities from foundation models to smaller, deployment-friendly variants.

You will work on distillation frameworks, model acceleration, and hardware-efficient inference across image/video generation and multimodal understanding, collaborating to deliver scalable solutions.

Qualifications

• BS in CS or related field or equivalent experience
• Expertise in efficient models and acceleration methods
• Experience with training generative AI or LLM models using PyTorch/JAX
• Strong communication and collaboration skills

Responsibilities

• Develop efficient algorithms and architectures for large-scale generative and multimodal models, using distillation/quantization to improve efficiency
• Advance scalable generative modeling approaches, including diffusion and autoregressive models, with focus on acceleration and efficiency

Skills

Large-scale models efficiency

PyTorch

JAX

Distillation frameworks

Model acceleration

Education

B.S. in Computer Science or related fields

Tools

PyTorch

JAX

Job description

TikTok’s Vision-Applied Research team seeks a Research Engineer/Scientist to design and implement efficient large-scale generative AI models, emphasizing distillation and compression to move capabilities from foundation models to smaller, deployment-friendly variants.

You will work on distillation frameworks, model acceleration, and hardware-efficient inference across image/video generation and multimodal understanding, collaborating to deliver scalable solutions.