Machine Learning Engineer - Model Training Infrastructure - USDS

$137,750 - $237,500/Yr

Tiktok - Seattle, WA

Full-time - Mid Level

Seattle, WA

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

The Machine Learning Engineer - Model Training Infrastructure role at TikTok focuses on designing and implementing a global-scale machine learning system for feeds, ads, and search ranking models. This position is part of the U.S. Data Security division, which emphasizes data protection and content assurance for U.S. users. The role involves enhancing the usability and flexibility of machine learning infrastructure, improving workflows for model training and serving, and mentoring interns.

Responsibilities

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

Requirements

• 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 C/C++/CUDA/Python with solid programming skills.
• Familiar with deep learning frameworks such as TensorFlow and PyTorch.
• Experience in developing and deploying large-scale systems.

Nice-to-haves

• Ability to work independently and complete projects from beginning to end in a timely manner.
• Good communication and teamwork skills to clearly communicate technical concepts with teammates.
• Experience improving core machine learning infrastructure (TensorFlow, PyTorch, and Jax).
• Experience contributing to an open-sourced machine learning framework (TensorFlow/PyTorch).
• Experience in using/designing open-source machine learning lifecycle management systems: TFX.

Benefits

• 100% premium coverage for employee medical insurance, approximately 75% for dependents, and a Health Savings Account (HSA) with company match.
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life, and AD&D insurance plans.
• Flexible Spending Account (FSA) options for Health Care, Limited Purpose, and Dependent Care.
• 10 paid holidays per year plus 17 days of Paid Personal Time Off (PPTO) and 10 paid sick days per year.
• 12 weeks of paid Parental leave and 8 weeks of paid Supplemental Disability.
• Mental and emotional health benefits through EAP and Lyra.
• 401K company match, gym, and cellphone service reimbursements.