Tech Lead, Machine Learning Infrastructure Engineer - USDS

Job summary

The Tech Lead, Machine Learning Infrastructure Engineer at TikTok USDS will drive the technical roadmap for large-scale distributed ML training and inferencing platforms, impacting various business domains.

Qualifications

• 5+ years of professional software engineering experience
• 3+ years of direct experience building and maintaining machine learning infrastructure
• Deep technical familiarity with core ML frameworks
• Strong system-level troubleshooting and debugging skills

Responsibilities

• Drive the technical roadmap for large-scale distributed ML training and inferencing platforms
• Architect and scale multi-node distributed training systems
• Build and scale low-latency, high-throughput model serving infrastructure
• Design robust, automated fault-detection systems
• Partner with Applied ML researchers and data platform teams
• Implement and enforce strict encryption-at-rest/in-transit controls

Skills

Degrees

Relocation