Drimva

TikTok

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

San Jose, California · Internship · Posted Oct 28, 2026

Education
Currently pursuing a PhD degree in artificial intelligence, computer science, operations research, automation, statistics, mathematics, or a related discipline.

About the job

Develop and evaluate LLMs, agents, and multimodal AI systems for e-commerce supply-chain and logistics workflows.

What you'll do

  • Build and improve domain LLMs, including post-training, reinforcement learning, and inference optimization for logistics tasks.
  • Develop multimodal understanding and agent capabilities, including retrieval, tool use, workflow automation, and integration with existing systems.
  • Support agent architecture, evaluation, monitoring, regression testing, and continuous improvement.

What they ask for

  • Currently pursuing a PhD 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 Python programming ability and familiarity with at least one of 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

  • Experience with agent systems, LLM post-training, benchmarks, or evaluation datasets.
  • Experience in e-commerce, logistics, operations research, data platforms, knowledge graphs, or enterprise knowledge management; relevant publications, open-source contributions, or projects.

Tools

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

AI focus

Applies LLMs, agents, reinforcement learning, retrieval, and multimodal AI to logistics and supply-chain tasks, including address correction, prediction, cost analysis, customer-service understanding, and workflow automation.

Model-written summary · the original posting is authoritative

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