Global Logistics AI Engineer - LLMs & Agent Systems

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

Job summary

TikTok Inc. is hiring into the E-commerce Global Supply Chain and Logistics team in San Jose. You will apply LLMs, agents, reinforcement learning, retrieval systems, and software engineering to real logistics problems at global scale, building AI-native capabilities for logistics operations.

As a graduate-friendly role, opportunities arise to pursue bold ideas and tackle complex challenges, unlocking growth while shaping scalable solutions across our global ecommerce network.

Qualifications

• PhD or recent PhD in AI, CS, ML, NLP, data mining or related discipline.
• Experience with LLMs, agents, RAG, tool use, post-training, evaluation or applied NLP systems.
• Strong Python programming and at least one production language (Java, C++, Go, or TS).
• Familiarity with ML frameworks (PyTorch, TensorFlow, JAX) and ecosystem tools.
• Strong problem decomposition, data analysis, and communication skills.

Responsibilities

• Develop large language models, agent systems and intelligent tech for global supply chain logistics.
• Build domain LLM capabilities for e-commerce logistics, including continual training and RL components.
• Develop multimodal understanding for logistics scenarios and integration with forecasting or decision systems.
• Create core agent capabilities for workflows, including RAG, context understanding, and tool use.
• Design robust agent architecture, memory/state management, observability and evaluation.

Skills

LLMs & agents

Tool use

NLP systems

Python

Production languages

Education

PhD in AI/CS/ML/NLP

Tools

PyTorch

TensorFlow

JAX

vLLM

HuggingFace

LangChain

LlamaIndex

Job description

TikTok Inc. is hiring into the E-commerce Global Supply Chain and Logistics team in San Jose. You will apply LLMs, agents, reinforcement learning, retrieval systems, and software engineering to real logistics problems at global scale, building AI-native capabilities for logistics operations.

As a graduate-friendly role, opportunities arise to pursue bold ideas and tackle complex challenges, unlocking growth while shaping scalable solutions across our global ecommerce network.