ML Infrastructure Engineer: Build Global ML Systems

TikTok USDS Joint Venture; San Jose (CA); On-site; USD 150,000 - 360,000; Full time

Benefits offered by this job

Health insurance

Dental insurance

Vision insurance

401(k) match

Parental leave

Disability coverage

Life insurance

Wellbeing benefits

Paid time off

Job summary

TikTok USDS Joint Venture LLC is seeking a Machine Learning Engineer to build and optimize a global-scale ML infrastructure for feeds, ads, and search ranking. You will design core ML components, optimize pipelines, and contribute to multi-tenant ML systems across a high-traffic platform.

The role emphasizes designing scalable ML infrastructure, mentoring teammates, and collaborating with cross-functional teams to advance real-time decision making and product quality.

Qualifications

• Bachelor's/Master's degree in Computer Science, Computer Engineering, or related fields or equivalent years of experience in a software engineering role.
• Proficient in at least one programming language such as Go/Python in Linux environment, with excellent coding skills.
• Familiar with open source distributed scheduling/orchestration/storage frameworks, such as Kubernetes (K8S), Yarn (Flink, MapReduce), Mesos, Celery, HDFS, Redis, S3, etc., with rich practical experience in machine learning system development.
• Experience in developing and deploying large-scale systems.

Responsibilities

• Design and implement a global-scale machine learning system for feeds, ads and search ranking models.
• Improve usability and flexibility of the machine learning infrastructure.
• Improve workflow of model training and serving, data pipelines, storage system and resource management for multi-tenancy ML systems.
• Design and develop key components of ML infrastructure and mentor interns.

Skills

Go/Python programming

Distributed ML systems

Large-scale system development

Education

Bachelor's/Master's in CS/CE or related

Tools

Kubernetes

Yarn

Flink

Mesos

Celery

HDFS

Redis

S3

Job description

TikTok USDS Joint Venture LLC is seeking a Machine Learning Engineer to build and optimize a global-scale ML infrastructure for feeds, ads, and search ranking. You will design core ML components, optimize pipelines, and contribute to multi-tenant ML systems across a high-traffic platform.

The role emphasizes designing scalable ML infrastructure, mentoring teammates, and collaborating with cross-functional teams to advance real-time decision making and product quality.