Research Engineer: Efficient Large-Scale GenAI Distillation

TikTok Inc.; San Jose (CA); On-site; USD 218,000 - 388,000; Full time

Benefits offered by this job

Medical insurance

Dental insurance

Vision insurance

401(k) with company match

Job summary

TikTok Inc. in California is seeking a Research Engineer/Scientist for the Vision-Applied Research team to design and implement efficient large-scale generative AI models with emphasis on distillation and compression.

You will help transfer capabilities from foundation models into smaller, faster systems, enabling scalable training, optimization, and deployment. The role involves developing distillation frameworks, model acceleration, and hardware-friendly inference, collaborating with

Qualifications

• B.S. in Computer Science or related field or equivalent experience.
• Expertise in efficient models with understanding of bottlenecks and acceleration methods.
• Proficiency in training generative AI or LLM models using PyTorch and JAX.
• Strong communication and collaboration skills in fast-paced environments.

Responsibilities

• Design and implement efficient algorithms and architectures for large-scale generative and multimodal models using distillation, quantization, and related methods.
• Advance scalable generative modeling approaches (diffusion, autoregressive) with focus on acceleration and efficiency.

Skills

B.S. in CS

Efficient models

PyTorch & JAX

Communication skills

Education

B.S. in CS or related field

Tools

PyTorch

JAX

Job description

TikTok Inc. in California is seeking a Research Engineer/Scientist for the Vision-Applied Research team to design and implement efficient large-scale generative AI models with emphasis on distillation and compression.

You will help transfer capabilities from foundation models into smaller, faster systems, enabling scalable training, optimization, and deployment. The role involves developing distillation frameworks, model acceleration, and hardware-friendly inference, collaborating with