Apply on Employer Site

TikTok · 2 weeks ago

Software Engineer - Model Training Infrastructure - USDS

Seattle, WA

Full-time

Onsite

Mid Level

$130K/yr - $246K/yr

TikTok is the leading destination for short-form mobile video, and they are seeking a Machine Learning Engineer - Model Training Infrastructure to support their AI infrastructure and recommendation platform. The role involves designing and implementing global-scale machine learning systems, improving workflow for model training, and mentoring interns.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo

No H1B

Responsibilities

Responsible for the design and implementation of a global-scale machine learning system for feeds, ads and search ranking models

Responsible for improving use-ability and flexibility of the machine learning infrastructure

Responsible for improving the workflow of model training and serving, data pipelines, storage system and resource management for multi-tenancy machine learning systems

Responsible for designing and developing key components of ML infrastructure and mentoring interns

Qualification

C/C++/CUDA/PythonDeep learning frameworksLarge-scale systemsMachine learning infrastructureOpen-source frameworksCommunication skillsTeamwork skills

Required

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, and have solid programming skills

Familiar with deep learning frameworks (TensorFlow/Pytorch)

Experience in developing and deploying large-scale systems