Machine Learning Engineer Intern (E-Commerce Supply Chain & Logistics-LLM/Agent) - 2027 Start (PhD)
Summary
PhD-level ML engineering internship at TikTok building LLM and agent capabilities for e-commerce supply chain and logistics. Day-to-day work includes fine-tuning LLMs with SFT/preference optimization, designing agent architectures, implementing RAG, and optimizing inference for address correction and trajectory prediction.
- Apply SFT and preference optimization
- Build domain LLM capabilities
- Build multimodal and structured understanding models
- Compress models and optimize inference latency
- Design agent architecture and runtime orchestration
- Develop agent workflows with task decomposition and execution loops
- Develop large language models for supply chain logistics
- Evaluate models using benchmarks and golden sets
- Implement Retrieval Augmented Generation and knowledge retrieval
- Implement memory and state management for agents
- Perform regression testing and bad case attribution
- Set up online feedback loops for continuous improvement
- Support address correction and trajectory prediction scenarios
- Train reinforcement learning for language models