Machine Learning Engineer, Recommendation - E-Commerce

$145,000 - $355,000/Yr

Tiktok - San Jose, CA

Full-time - Mid Level

San Jose, CA

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

About the position

TikTok is the leading destination for short-form mobile video, and our mission is to inspire creativity and bring joy. As part of our E-commerce team, we are focused on connecting customers with excellent sellers and quality products through innovative solutions like live-streaming and short videos. The E-commerce Recommendation and Marketing team is composed of applied machine learning engineers and research scientists dedicated to enhancing the shopping experience for TikTok users by developing and scaling our recommendation systems. In this role, you will be responsible for designing and building optimization algorithm strategies for our large-scale e-commerce recommendation algorithm pipeline, which handles millions of products and creators' content. You will create user interest models, analyze vast amounts of data, and design algorithms that efficiently explore users' latent interests. Your work will involve developing predictive models for candidate generation and ranking, including Click Through Rate and Conversion Rate predictions, and you will be expected to build real-time data pipelines and innovate in model optimization. You will also design and build supporting tools for debugging and analysis as needed. This position is crucial for ensuring that our recommendation systems provide a stable and high-quality shopping experience for our users, ultimately driving impact for our company and the communities we serve.

Responsibilities

• Responsible for the build and design of optimization algorithm strategies for large-scale e-commerce recommendation algorithm pipeline.
• Build long and short term user interest models, analyze and extract relevant information from large amounts of various data and design algorithms to explore users' latent interests efficiently.
• Design, develop, evaluate and iterate on predictive models for candidate generation and ranking, including Click Through Rate and Conversion Rate prediction.
• Build real-time data pipelines, feature engineering, model optimization and innovation.
• Design and build supporting/debugging tools as needed.

Requirements

• Bachelor's degree in computer science or relevant areas.
• 3+ years of experience with a solid foundation in data structure and algorithm design.
• Proficient in using one of the programming languages such as Python, Java, C++, R, etc.
• Familiar with common machine/deep learning, causal inference, and operational optimization algorithms, including classification, regression, clustering methods, as well as mathematical programming and heuristic algorithms.
• Familiar with at least one framework of TensorFlow / PyTorch / MXNet and its training and deployment details, as well as training acceleration methods such as mixed precision training and distributed training.
• Familiar with big data related frameworks and applications, preferably with MR or Spark.

Nice-to-haves

• Experience in recommendation systems, online advertising, ranking, search, information retrieval, natural language processing, machine learning, large-scale data mining, or related fields.
• Publications at KDD, NeurlPS, WWW, SIGIR, WSDM, ICML, IJCAI, AAAI, RECSYS and related conferences/journals.
• Experience in data mining/machine learning competitions such as Kaggle/KDD-cup.

Benefits

• 100% premium coverage for employee medical insurance, approximately 75% premium coverage for dependents.
• Health Savings Account (HSA) with a company match.
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life and AD&D insurance plans.
• Flexible Spending Account (FSA) Options like 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.