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

TikTok logo

TikTok

Inspire creativity, bring joy.

10,001+ employeesLos Angeles, CaliforniaEntertainment Providers
Actively hiringPaidEntertainment Providers
Apply Now

Best applied by Nov 14

Access our internships job board to view 10,000+ active internships.


About the role#

The E-commerce Global Supply Chain and Logistics team builds AI-native capabilities for our global operations. We focus on logistics agents, address intelligence, context engineering, and workflow automation. This internship is for candidates interested in applying LLMs, reinforcement learning, and retrieval systems to logistics problems at a global scale. You will contribute to our research and products while working with industry experts.

What you'll do#

  • Support the development of large language models and agent systems for global e-commerce logistics.
  • Build domain-specific LLM capabilities, including continued pre-training, SFT, preference optimization, and reinforcement learning (GRPO/PPO).
  • Design reward or judge models, optimize inference costs, and work on scenarios like address correction, trajectory prediction, and root-cause analysis.
  • Develop multimodal understanding for logistics, integrating text, numerical time series, product attributes, and documents to support decision systems.
  • Create core agent capabilities such as task decomposition, RAG, knowledge retrieval, and tool use.
  • Manage agent architecture, including runtime orchestration, memory management, model routing, and observability.
  • Build evaluation frameworks, including benchmark sets, regression testing, and online feedback loops.

What you'll need#

  • Currently pursuing a PhD in artificial intelligence, computer science, operations research, automation, statistics, mathematics, or a related field.
  • Experience with LLMs, agents, RAG, tool use, post-training, or applied NLP systems.
  • Strong Python programming skills and familiarity with at least one production language like Java, C++, Go, or TypeScript.
  • Proficiency with frameworks such as PyTorch, TensorFlow, JAX, vLLM, Hugging Face, LangChain, or LlamaIndex.
  • Research or project experience in coding agents, workflow orchestration, or automated evaluation systems.
  • Experience with LLM post-training techniques like SFT, DPO, PPO, GRPO, RLHF, or RLAIF.
  • Experience building benchmarks from real user data, including taxonomy design and error attribution.
  • Background in e-commerce, logistics, knowledge graphs, or data platforms is a plus.
  • A portfolio of published papers, open-source contributions, or hands-on projects in AI engineering or machine learning systems.

Location & details#

  • Location: San Jose, California.
  • Modality: On-site.
  • Employment: Full-time internship.
  • Compensation: $60 per hour.
  • Status: Paid position.
  • Hiring process: Applications are reviewed on a rolling basis.

About TikTok

TikTok operates as an entertainment provider. The company maintains a global presence with offices in cities including Los Angeles, Singapore, New York, London, and Tokyo. It employs over 92,000 people. The organization is privately held.

LinkedInWebsiteCareers emailTikTok contact channels unlock with a free account
All TikTok internships

How to get in at TikTok

Landing an internship at TikTok requires speed because early applicants often get seen before the pile grows. Intern Insider sends an instant alert the moment a role matching your target is published anywhere, so you can apply among the first. This gives you a better chance of your resume being reviewed while the team is still actively filling the position. You should aim to be one of the first people to submit your materials. Reaching out to the right person can also improve your chances of getting a response. Intern Insider surfaces the recruiters behind the company roles so you can contact them directly to ask about the role or a referral. This direct approach often works better than waiting for a response from a general application queue.

Get alerts for new TikTok internships
Share

Interns who landed roles with Intern Insider

“The intern insider job board is an extremely useful resource; especially for students just beginning their career journeys. Besides the amazing catalogue of new postings, there are so many different tools to better set yourself up for success.”
Jordan L., Business & Computer Science Student

Jordan L.

Business & Computer Science Student

“I didn't realize how much time I was wasting until I tried Intern Insider. Instead of scrolling for hours, I just apply directly through the platform.”
Jemma V., Private Wealth Intern

Jemma V.

Private Wealth Intern

Apply to more than this one.

10,000+ open internships in one place, with the deadlines and the people to email.

Create free account
Apply by Nov 14Apply Now
Privacy

Cookies

Manage optional cookies and similar technologies. You can update these choices anytime. Learn more

EssentialKeep you signed in and support security and basic site features.Always on
AnalyticsHelp us improve the site by measuring how people use it.
AdvertisingHelp us show relevant ads and measure how well they work.