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