Tech Lead, AI Infra Compute

Tiktok • Seattle, WA

60d

About The Position

Compute division focuses on building large-scale and highly available cloud infrastructure, which supports both public cloud products (like VolcEngine ECS service) and the internal products. The Compute-US team focuses on the development and research of the compute infrastructure platform.

Requirements

• BS or above degree in Computer Science or related technical field or equivalent practical experience.
• 5+ years experience of AI and AI Infra related projects.
• 3+ years experience as a tech lead.

Nice To Haves

• Deep understanding of cloud infrastructure or AI infrastructure.
• Good communication, self-motivation, engineering practice, and working collaboratively in cross-functional teams.
• Familiar with at least two of the areas below:
• GPU Infra (GPU cluster management, job scheduling, collective communication libraries, etc)
• LLM Inference engines, including performance optimizations and cost optimizations.
• Machine learning frameworks such as PyTorch or other frameworks.
• AI Infra hardware, including next-gen CPU/GPU, training/inference server platforms, data center infra, etc.

Responsibilities

• Lead the design and prototype of key technologies or products.
• Research and follow trends from industry companies, open-source communities and academia.
• Review design and implementation and ensure it is aligned with long term vision and roadmap.
• Communicate and coordinate with partner teams to deliver projects on time with high quality.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Industry: Broadcasting and Content Providers
• Number of Employees: 5,001-10,000 employees