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

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

Job summary

TikTok is hiring for an AI/ML-focused role within its E-commerce Global Supply Chain and Logistics team. You will work on AI-native capabilities for global logistics, including agents, address intelligence, and workflow automation to solve real-world supply chain problems at scale.

Graduates will have opportunities to pursue bold ideas, tackle complex challenges, and grow with the team. Applicants should include onboarding availability and graduation dates in their resume.

Qualifications

• Bachelor's degree in AI, CS, OR, automation, statistics or related field.
• Proficient in ML and statistics with practical application experience.
• Proficient in Python/Java and one ML/DL framework (TensorFlow/PyTorch).
• Strong data analysis, curiosity, initiative and teamwork.

Responsibilities

• Develop deep learning and operations research models for global supply chain and logistics.
• Predict ETA and logistics events using big data and DL models; build knowledge graphs and spatio-temporal predictions.
• Apply time series forecasting to demand, warehouse and manpower planning.
• Contribute to growth algorithms for e-commerce merchants/creators and collaborate with operations and product teams.
• Optimize traffic distribution and pricing/subsidy strategies using data-driven methods.

Skills

Machine learning

Statistics

Python/Java

Data analysis

Communication

Education

Bachelor's degree in AI/CS/OR/related discipline

Master's degree in Software Development/CS/CE or related discipline

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

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).

• Employees have day one access to medical, dental, and vision insurance.
• Employees have a 401(k) savings plan with company match.
• Employees have paid parental leave.
• Employees have short-term and long-term disability coverage.
• Employees have life insurance.
• Employees have wellbeing benefits.
• Employees also receive 10 paid holidays per year.
• Employees also receive 10 paid sick days per year.
• Employees also receive 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:

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

About TikTok

TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and we also have offices in New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.

Why Join Us

Inspiring creativity is at the core of TikTok's mission. Our innovative product is built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and bring joy - a mission we work towards every day.

We strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. Every challenge is an opportunity to learn and innovate as one team. We're resilient and embrace challenges as they come. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our company, and our users. When we create and grow together, the possibilities are limitless. Join us.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

TikTok Accommodation

TikTok is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at https://tinyurl.com/RA-request