Machine Learning Engineer Graduate (E-Commerce Supply Chain & Logistics) - 2027 Start

TikTok Inc.; San Jose (CA); On-site; USD 128,000 - 317,000; Full time

Benefits offered by this job

Medical, dental, vision insurance

401(k) with company match

Paid parental leave

Disability coverage

Life insurance

Wellbeing benefits

Paid holidays

Job summary

TikTok Inc. is building an AI-native logistics stack for the global E-commerce supply chain. You will develop deep learning and OR models to optimize ETA, events, and network knowledge graphs, while applying NLP and ML methods to address address verification and optimization challenges at scale.

You will forecast demand, manpower, and inventory, and work with merchants/creators to drive high‑quality growth, leveraging large‑scale data and production-ready ML systems.

Qualifications

• Bachelor's degree in AI/CS/OR/Math or related field.
• Strong ML and statistics knowledge with practical application.
• Proficient in Python/Java and common ML frameworks (TensorFlow/PyTorch).
• Strong data analysis, initiative, teamwork and communication skills.

Responsibilities

• Develop deep learning and operations research models for global E‑Commerce logistics.
• Use ML/DL to predict ETA and logistics events to improve user experience.
• Apply time-series forecasting for demand, manpower, and inventory planning.
• Contribute to growth algorithms for merchants/creators and coordinate with operations and product teams.
• Leverage data mining and ML to optimize traffic, pricing, and subsidies strategies.

Skills

Machine learning

Statistics

Python/Java

Data analysis

Teamwork

Education

Bachelor's degree in AI/CS/OR/Math

Master's degree in Software Development/CS/Engineering

Tools

TensorFlow

PyTorch

Job description

Responsibilities

Join the E-commerce Global Supply Chain and Logistics team at TikTok. We are building AI-native capabilities for global logistics, including logistics agents, address intelligence, context engineering, agent evaluation, and workflow automation for complex supply chain operations. This role is for candidates who want to apply LLMs, agents, reinforcement learning, retrieval systems, and software engineering to real logistics problems at global scale.

We are looking for talented individuals to join our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.

Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation date clearly in your resume.

• Responsible for the development of deep learning and operations research models and related intelligent systems for the supply chain and logistics of the global E-Commerce business.
• Utilize e-commerce big data and deep learning models to predict end-to-end estimated time of arrival (ETA), and some logistics events such as failed delivery, delivered but not received to enhance the user logistics experience. Build logistics network knowledge graphs and predict the spatio-temporal trajectory sequence of express packages through deep learning, statistical inference and other algorithmic methods. Use NLP and LLM algorithms to handle address problems such as address verification and address suggestion.
• Utilize time series forecasting techniques to predict sales at different granularities and horizons, such as warehouse-level manpower forecasting, inventory-level demand forecasting etc. We need strong machine learning and deep learning skills to detect important factors and model the relationship between the future and history.
• Participate in the growth of e-commerce merchants/creators, responsible for the growth algorithm of e-commerce merchants/creators, including potential merchants/creators mining algorithm, tiering algorithm, out-reach algorithm, growth algorithm, etc. Work closely with e-commerce merchants/creators related operations and product teams to achieve the high-quality growth goal of e-commerce merchants/creators.
• Combined with massive e-commerce information, through data mining and machine learning, predict the performance of products in traffic/conversion, improve the efficiency of product operation strategy and pricing/subsidy strategy, and assist in optimizing traffic distribution, to satisfy the requirements of product growth, cost optimization and some other scenarios.

Qualifications

Minimum Qualifications

• Individuals who are completing or have recently completed a Bachelor's degree in artificial intelligence, computer science, operations research, automation, statistics, mathematics, or a related discipline
• Proficient in machine learning and statistics, familiar with the principles and model structures of common machine learning and deep learning algorithms for classification, regression, and clustering, with practical application experience.
• Proficient in programming languages such as Python/Java, possessing excellent coding abilities, and familiar with at least one common machine learning/deep learning framework (TensorFlow/PyTorch etc.).
• Demonstrates a spirit of exploration, outstanding data analysis capabilities, strong initiative, and excellent teamwork and communication skills.

Preferred Qualifications

• Individuals who are completing or have recently completed a Master's degree in Software Development, Computer Science, Computer Engineering or a related discipline.
• Hands-on experience with coding agents, long-horizon agents, computer-use agents, agent harnesses, workflow orchestration, or automated evaluation frameworks.
• Experience with LLM post-training, including SFT, DPO, PPO, GRPO, RLHF, RLAIF, reward modeling, counterfactual data, or evidence-driven decision training.
• Experience building benchmarks from real production questions, including taxonomy design, golden answer construction, error attribution, and regression evaluation.
• Experience with logistics, e-commerce, operations research, data platforms, knowledge graphs, or enterprise knowledge management systems.
• Published papers or strong open-source work in LLMs, agents, NLP, data mining, machine learning systems, evaluation, or AI engineering.

Job Information

[For Pay Transparency] Compensation Description (annually)

The base salary range for this position in the selected city is $128000 - $316800 annually.

Compensation may vary outside of this range depending on a number of factors, including a candidate's qualifications, skills, competencies and experience, and location. Base pay is one part of the Total Package that is provided to compensate and recognize employees for their work, and this role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.

Benefits may vary depending on the nature of employment and the country work location. Employees have day one access to medical, dental, and vision insurance, a 401(k) savings plan with company match, paid parental leave, short-term and long-term disability coverage, life insurance, wellbeing benefits, among others. Employees also receive 10 paid holidays per year, 10 paid sick days per year and 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).

The Company reserves the right to modify or change these benefits programs at any time, with or without notice.

For Los Angeles County (unincorporated) Candidates

Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act. Our company believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment:

1. Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;
2. Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems;
3. Exercising sound judgment.