Drimva

TikTok

Machine Learning Engineer Graduate (E-Commerce Supply Chain & Logistics - LLM/Agent) - 2027 Start (PhD)

San Jose, California · Full-time · Posted Oct 28, 2026

Education
PhD in a listed or related discipline; no equivalent-experience alternative stated.

About the job

Develops LLMs, agents, and intelligent systems for global e-commerce supply chain and logistics, from model training and multimodal understanding to workflow automation and evaluation.

What you'll do

  • Develop and apply LLMs and agent systems to logistics and supply chain problems, including model training, optimization, and business applications.
  • Build multimodal and structured understanding capabilities for logistics data and connect them to forecasting or decision systems.
  • Design agent architecture and engineering systems for workflow execution, retrieval, evaluation, monitoring, and continuous improvement.

What they ask for

  • Completing or recently completed a 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.
  • Strong Python programming and proficiency in at least one production-oriented language such as Java, C++, Go, or TypeScript.
  • Familiarity with machine learning and deep learning frameworks or ecosystems such as PyTorch, TensorFlow, JAX, vLLM, Hugging Face, LangChain, or LlamaIndex.

Nice to have

  • Hands-on experience with coding agents, long-horizon agents, computer-use agents, agent harnesses, workflow orchestration, or automated evaluation.
  • Experience in LLM post-training, production benchmarks, logistics, e-commerce, operations research, data platforms, knowledge graphs, or enterprise knowledge management.

Tools

Python, Java, C++, Go, TypeScript, PyTorch, TensorFlow, JAX

AI focus

Builds and applies LLMs, agents, retrieval systems, reinforcement learning, multimodal understanding, and automated evaluation for supply chain and logistics workflows.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)