Apply on Employer Site

TikTok · 3 weeks ago

Tech Lead, AI Infra Compute

Seattle, WA

Full-time

Onsite

Senior Level

$198K/yr - $416K/yr

5+ years exp

TikTok is the leading destination for short-form mobile video, dedicated to inspiring creativity and bringing joy. The Tech Lead, AI Infra Compute will lead the design and prototype of key technologies while researching industry trends and coordinating with partner teams to ensure project delivery. This role requires a strong background in AI and cloud infrastructure, along with leadership experience.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo

H1B Sponsor Likely

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

Qualification

AI Infra experienceTech lead experienceCloud infrastructure knowledgeGPU Infra managementMachine learning frameworksAI Infra hardwareCommunicationSelf-motivationCollaborative teamwork

Required

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

Preferred

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

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)

Company

TikTok

Glassdoor3.4

TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.

Founded in 2003

Los Angeles, California, USA

10001+ employees

https://www.tiktok.com

H1B Sponsorship

TikTok has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (979)

2024 (601)

2023 (387)

2022 (322)

2021 (133)

2020 (72)