Generative AI Research Engineer — Model Distillation

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

Job summary

TikTok is seeking a Research Engineer/Scientist to design efficient large-scale generative AI models and distillation methods. You will work on transferring capabilities from foundation models into smaller, efficient models for scalable training, optimization, and deployment.

Applicants should have strong experience with PyTorch/JAX and a track record in model compression, diffusion or autoregressive techniques, and collaboration in fast-paced teams.

Qualifications

• B.S. in Computer Science or related field or equivalent experience.
• Experience with efficient models and acceleration techniques.
• Comfort with PyTorch and JAX for training generative AI models.

Responsibilities

• Design and implement efficient algorithms for large-scale generative and multimodal models.
• Develop distillation frameworks and acceleration methods for model compression.

Skills

PyTorch

JAX

Model compression

Distillation

Large-scale models

Communication

Collaboration

Optimization

Education

B.S. in Computer Science or related fields

Job description

TikTok is seeking a Research Engineer/Scientist to design efficient large-scale generative AI models and distillation methods. You will work on transferring capabilities from foundation models into smaller, efficient models for scalable training, optimization, and deployment.

Applicants should have strong experience with PyTorch/JAX and a track record in model compression, diffusion or autoregressive techniques, and collaboration in fast-paced teams.