Machine Learning Engineer Intern (E-Commerce Supply Chain & Logistics-LLM/Agent)

TikTok
Seattle, WA
Job Description
Role Overview

Join the E-commerce Global Supply Chain and Logistics team at TikTok to contribute to AI-native capabilities for global logistics, including logistics agents, address intelligence, context engineering, agent evaluation, and workflow automation for complex supply chain operations.

What You Will Do

Support the development of large language models, agent systems, and related intelligent systems for the supply chain and logistics of the global E-Commerce business, including building domain LLM model capabilities, multimodal and structured understanding capabilities, and core agent capability development.

Why It Might Be a Fit

This role is ideal for PhD students who want to apply LLMs, agents, reinforcement learning, retrieval systems, and software engineering to real logistics problems at global scale, with opportunities for hands-on learning, community-building, and professional development.

Requirements

  • Currently pursuing a PhD degree in artificial intelligence, computer science, operations research, automation, statistics, mathematics, or a related discipline
  • Experience with LLMs, agents, RAG, tool use, post-training, evaluation, or applied NLP systems
  • Strong programming ability in Python and familiarity with at least one production-oriented language such as Java, C++, Go, or TypeScript
  • Familiarity with machine learning and deep learning frameworks such as PyTorch, TensorFlow, JAX, vLLM, Hugging Face, LangChain, LlamaIndex, or similar ecosystems
]]>