Large Model Training Acceleration Engineer

TikTok
San Jose, CA
Job Description
We are seeking an experienced AI model optimization engineer to work at the cutting edge of AI efficiency, enhancing the performance, scalability, and deployment of large-scale generative AI models.

Requirements

  • Optimize large model training pipelines to improve efficiency, speed, and scalability.
  • Develop and improve distributed training strategies such as data parallelism, model parallelism, pipeline parallelism and communication to accelerate model training.
  • Benchmark and profile deep learning models to identify performance bottlenecks and optimize computational resources.

Benefits

  • Medical, dental, and vision insurance
  • 401(k) savings plan with company match
  • Paid parental leave
  • Short-term and long-term disability coverage
  • Life insurance
  • Wellbeing benefits
  • 10 paid holidays per year
  • 10 paid sick days per year
  • 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure)
]]>