Graduate ML Engineer - LLMs for Global Logistics

TikTok; Seattle (WA); On-site; USD 154,000 - 301,000; Full time

Job summary

TikTok is hiring for a position in Seattle to advance AI-native logistics across the global e-commerce network. You will develop large language models, agent systems, and associated intelligent tooling for complex supply chain operations, including RAG, tool use, and evaluation.

Ideal candidates will hold a PhD in relevant fields and demonstrate strong Python and production-language skills, with experience in PyTorch, TensorFlow, JAX, and modern NLP ecosystems.

Qualifications

• PhD in artificial intelligence, computer science, machine learning, natural language processing, data mining, software engineering or a related discipline.
• Experience with LLMs, agents, RAG, tool use, post-training, evaluation, or applied NLP systems, with the ability to connect model behavior to business and engineering requirements.
• Strong programming ability in Python and at least one production-oriented language such as Java, C++, Go, or TypeScript; comfortable building end-to-end systems, not only model experiments.
• Familiarity with machine learning and deep learning frameworks such as PyTorch, TensorFlow, JAX, vLLM, Hugging Face, LangChain, LlamaIndex, or similar ecosystems.
• Strong problem decomposition skills, data analysis capability, and communication skills; able to work with ambiguous business questions and convert them into executable technical plans.

Responsibilities

• Responsible for the development of large language models, agent systems, and related intelligent systems for the supply chain and logistics of the global E-Commerce business.
• Build and land domain LLM model capabilities for e-commerce supply chain and logistics, covering continued pre-training / CPT, SFT, preference optimization, reinforcement learning such as GRPO / PPO, reward or judge model design, model compression, inference cost and latency optimization, and business applications such as address correction, trajectory prediction, logistics cost analysis, customer service semantic understanding, and root-cause analysis.
• Develop multimodal and structured understanding capabilities for logistics and supply chain scenarios, focusing on unified modeling of text, numerical time series, events, product attributes, images, and documents to support scenario simulation, explainable prediction, and integration with existing forecasting or decision systems.
• Build core agent capabilities for team and business workflows, including AutoResearch for new solution exploration, Harness-based task decomposition and execution loops, RAG and knowledge retrieval, context understanding, skill / tool use, evidence grounding, and Clone & Adapt workflows for reusing proven solutions across markets and logistics scenarios.
• Design and improve agent architecture and engineering systems, including runtime orchestration, memory and state management, model / tool routing, permission-safe execution, observability, benchmark and Golden Set evaluation, badcase attribution, regression testing, online feedback loops, and continuous evolution mechanisms that improve context, skills, workflows, and model behavior over time.

Skills

Python

LLMs

End-to-end systems

Agent systems

NLP

Education

PhD in AI/CS/NLP

Tools

PyTorch

TensorFlow

JAX

LangChain

LlamaIndex

HuggingFace

Job description

TikTok is hiring for a position in Seattle to advance AI-native logistics across the global e-commerce network. You will develop large language models, agent systems, and associated intelligent tooling for complex supply chain operations, including RAG, tool use, and evaluation.

Ideal candidates will hold a PhD in relevant fields and demonstrate strong Python and production-language skills, with experience in PyTorch, TensorFlow, JAX, and modern NLP ecosystems.