Generative AI Research Engineer

TikTok; San Jose (CA); On-site; USD 150,000 - 210,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, with emphasis on distillation and compression. You will prototype methods and infrastructure to transfer capabilities from foundation models into smaller, scalable systems.

You will work on distillation frameworks, model acceleration, and hardware-efficient inference, collaborating with researchers to push state-of-the-art in image/video generation and

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/LLM models using PyTorch and JAX.
• Strong communication and collaboration skills in fast-paced environments.

Responsibilities

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

Skills

Efficient models

Communication

Collaboration

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, with emphasis on distillation and compression. You will prototype methods and infrastructure to transfer capabilities from foundation models into smaller, scalable systems.

You will work on distillation frameworks, model acceleration, and hardware-efficient inference, collaborating with researchers to push state-of-the-art in image/video generation and