Principal ML Engineer - Large Scale Training Performance Optimization
职位介绍
Lead distributed training of large-scale models across multi-GPU systems to convergence.
Design and optimize end-to-end training pipelines, leveraging data, tensor, pipeline, and expert parallelism (including ZeRO) to scale out.
Implement algorithmic and kernel-level optimizations to improve training throughput and efficiency.
Stay current with the latest training methods and contribute changes to open-source projects.
Collaborate across teams and stakeholders to influence the direction of the AI platform while communicating results clearly.
Requires a master's or PhD in CS/AI with hybrid (partial on-site) work in San Jose, CA or Bellevue, WA.
Design and optimize end-to-end training pipelines, leveraging data, tensor, pipeline, and expert parallelism (including ZeRO) to scale out.
Implement algorithmic and kernel-level optimizations to improve training throughput and efficiency.
Stay current with the latest training methods and contribute changes to open-source projects.
Collaborate across teams and stakeholders to influence the direction of the AI platform while communicating results clearly.
Requires a master's or PhD in CS/AI with hybrid (partial on-site) work in San Jose, CA or Bellevue, WA.
查看完整职位
工作职责、任职要求、技能与福利 — 免费创建账号即可查看。
已有账户？ 登录
相似职位
其他可能适合您的职位。
? Intern, Machine Learning Engineer ? Senior Data Scientist - Product Analytics (Payments/Ecommerce) ? Data Scientist Graduate (TikTok-Product-Data Science)-2026 Start (PhD) ? [HackerRank] Software Engineer Graduate (On-Device AI) - 2025 Start (BS/MS) ? Senior Director, Risk Data Science ? Data Scientist Graduate (TikTok-Product-Data Science)-2026 Start (BS/MS)