TikTok
Posted 2h ago

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

TikTok
Seattle or Los Angeles or Singapore or New York City or London or Dublin or Paris or Berlin or Dubai or Jakarta or Seoul or Tokyo
OnsiteInternship
Responsibilities
  • developing models
  • building agents
  • evaluating systems
Requirements
  • PhD candidate in AI
  • Computer science
  • Operations research
  • Automation
  • Statistics
  • Mathematics, or related field
  • LLM, agent, RAG, or NLP experience
  • Strong Python and production-language programming skills
Technical tools mentioned
PythonJavaC++GoTypeScriptPyTorchTensorFlowJAXvLLMHugging FaceLangChainLlamaIndexRAGSFTDPOPPOGRPORLHFRLAIFAutoResearchHarnessLlamaIndex

Job description

Join the E-commerce Global Supply Chain and Logistics team at TikTok. We are building AI-native capabilities for global logistics, including logistics agents, address intelligence, context engineering, agent evaluation, and workflow automation for complex supply chain operations. This role is for candidates who want to apply LLMs, agents, reinforcement learning, retrieval systems, and software engineering to real logistics problems at global scale.

We are looking for talented individuals to join us for an internship. PhD internships at Our Company provide students with the opportunity to actively contribute to our products and research, as well as to the organization's future plans and emerging technologies.
Our dynamic internship experience blends hands-on learning, enriching community-building and professional development events, and collaboration with industry experts.
Applications will be reviewed on a rolling basis, so we encourage you to apply early. Please clearly state your availability in your resume (Start date, End date).

Responsibilities:
- Responsible for supporting the development of large language models, agent systems, and related intelligent systems for the supply chain and logistics of the global E-Commerce business.
- Participate in building domain LLM model capabilities for e-commerce supply chain and logistics, including continued pre-training / CPT, SFT, preference optimization, reinforcement learning such as GRPO / PPO, reward or judge model design, model compression, inference cost and latency optimization, and landing scenarios such as address correction, trajectory prediction, logistics cost analysis, customer service semantic understanding, and root-cause analysis.
- Support multimodal and structured understanding capabilities for logistics and supply chain scenarios, focusing on unified modeling of text, numerical time series, events, product attributes, images, and documents to support scenario simulation, explainable prediction, and integration with existing forecasting or decision systems.
- Support core agent capability development for team and business workflows, including AutoResearch, Harness-based task decomposition and execution loops, RAG and knowledge retrieval, context understanding, skill / tool use, evidence grounding, and Clone & Adapt workflows for reusing proven solutions across markets and logistics scenarios.
- Support agent architecture, engineering, evaluation, and evolution work, including runtime orchestration, memory and state management, model / tool routing, permission-safe execution, observability, benchmark and Golden Set evaluation, badcase attribution, regression testing, online feedback loops, and continuous improvement of context, skills, workflows, and model behavior.

Minimum Qualifications:
- Currently pursuing a PhD degree 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 programming ability in Python and familiarity with at least one production-oriented language such as Java, C++, Go, or TypeScript.
- Familiarity with machine learning and deep learning frameworks such as PyTorch, TensorFlow, JAX, vLLM, Hugging Face, LangChain, LlamaIndex, or similar ecosystems.

Preferred Qualifications:
- Research or project experience in coding agents, long-horizon agents, computer-use agents, agent harnesses, workflow orchestration, or automated evaluation systems.
- Experience with LLM post-training, including SFT, DPO, PPO, GRPO, RLHF, RLAIF, reward modeling, counterfactual data, or evidence-driven decision training.
- Experience building benchmarks or evaluation datasets from real user questions, including taxonomy design, golden answers, error attribution, and regression testing.
- Experience with e-commerce, logistics, operations research, data platforms, knowledge graphs, or enterprise knowledge management systems.
- Published papers, strong open-source contributions, or hands-on projects in LLMs, agents, NLP, data mining, machine learning systems, evaluation, or AI engineering.

TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and we also have offices in New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.



Inspiring creativity is at the core of TikTok's mission. Our innovative product is built to help people authentically express themselves, discover and connect – and our global, diverse teams make that possible. Together, we create value for our communities, inspire creativity and bring joy - a mission we work towards every day.

We strive to do great things with great people. We lead with curiosity, humility, and a desire to make impact in a rapidly growing tech company. Every challenge is an opportunity to learn and innovate as one team. We're resilient and embrace challenges as they come. By constantly iterating and fostering an "Always Day 1" mindset, we achieve meaningful breakthroughs for ourselves, our company, and our users. When we create and grow together, the possibilities are limitless. Join us.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

f:

TikTok is committed to providing reasonable accommodations in our recruitment processes for candidates with disabilities, pregnancy, sincerely held religious beliefs or other reasons protected by applicable laws. If you need assistance or a reasonable accommodation, please reach out to us at https://tinyurl.com/RA-request

About TikTok

Global short-form video hosting and social media platform.

Year founded
2016
Employees
10000
Organization type
Private
Latest investment
Raised $9.40B Private Equity (2020) — led by SoftBank, KKR, General Atlantic
Subsidiaries
Headquarters
US

Similar jobs

Machine Learning Engineer roles near Seattle, Washington
22h
Save
Mark Applied
Hide
Machine Learning Engineer - On-Device Adaptive Control
Seattle or West
OnsiteFull Time
Apple
AppleNASDAQ: AAPL: Designs and sells consumer electronics, software, and online services.
Requires an MS or PhD in a relevant technical field, or a BS with relevant experience; expertise in control or reinforcement learning, Python, C/C++, sensor data, and end-to-end prototyping.
Python, C, C++
1d
Save
Mark Applied
Hide
Machine Learning Engineer Graduate (E-Commerce Risk Control)- 2027 Start
Seattle, Washington, United States
OnsiteFull Time
ByteDance
ByteDance: Developing AI-driven content platforms and mobile applications.
0+ YOEBachelor's degree in computer science, computer engineering, or related field; modern machine learning proficiency; Python, SQL/Hive, and Hadoop experience; strong algorithms, data structures, and problem-solving skills.
Python, SQL, Hive, Hadoop, ReAct
1d
Save
Mark Applied
Hide
Staff Machine Learning Engineer, Causal Inference
San Francisco or Sunnyvale or Los Angeles or Seattle or New York City
$204k-$299k/yr OnsiteFull Time
DoorDash
DoorDashNASDAQ: DASH: On-demand delivery platform connecting consumers with local merchants.
Deep causal inference, econometrics, experimentation, or causal ML experience; production ML engineering; rigorous model evaluation; and ability to collaborate across engineering, analytics, product, and business teams.
ML, CUPED, Covey, Covey Scout for Inbound
2d
Save
Mark Applied
Hide
Senior Machine Learning Engineer
Seattle, Washington, United States
$151k-$241k/yr HybridFull Time
Axon
AxonNASDAQ: AXON: Develops weapons and software for law enforcement and safety.
6+ YOEBachelor's degree in a technical field, 6+ years of software engineering experience deploying AI/ML models to production, cloud architecture, infrastructure as code, Python, C++, ML frameworks, and Linux expertise.
Python, C++, PyTorch, TensorFlow, Linux
3d
Save
Mark Applied
Hide
Senior ML Engineer
Bellevue or Palo Alto or San Francisco or New York City
$149k-$224k/yr HybridFull Time
Salesforce
SalesforceNYSE: CRM: Sells cloud-based customer relationship management and business software solutions.
Architects threat-detection strategies, formulates security threats mathematically, develops probabilistic and graph analytics capabilities, and operationalizes production machine learning models.
AI, Data, CRM, Agentforce
3d
Save
Mark Applied
Hide
Principal Machine Learning Engineer
San Jose or Seattle or San Francisco
$206k-$379k/yr OnsiteFull Time
Adobe
AdobeNASDAQ: ADBE: Provides software for digital media creation and marketing analytics
8+ YOERequires 8+ years in production-scale machine learning engineering, 3+ years leading GPU-intensive GenAI inference systems, and expertise in inference tools, APIs, distributed systems, and technical leadership.
PyTorch, CUDA, Triton, TensorRT, Nvidia Dynamo, Python, Kubernetes, MLOps
6d
Save
Mark Applied
Hide
Staff Machine Learning Engineer, Generative AI Modeling and Inference
Los Angeles or Seattle or Palo Alto or New York City or Bellevue
$195k-$343k/yr OnsiteFull Time
Snap
SnapNYSE: SNAP: Develops social media applications and augmented reality technology.
8+ YOEBachelor's degree or equivalent experience and 8+ years of post-bachelor's ML experience, or advanced degree with equivalent experience. Requires computer vision or generative modeling and ML framework experience.
TensorFlow, PyTorch, JAX, MLX, scikit-learn, GPU, CPU, NPU, RSUs
6d
Save
Mark Applied
Hide
Staff Machine Learning Engineer, Generative AI Modeling and Inference
Los Angeles or Seattle or Palo Alto or New York City or Bellevue
$229k-$343k/yr OnsiteFull Time
Snap
SnapNYSE: SNAP: Provides visual messaging software and augmented reality wearable devices.
8+ YOEBachelor's degree or equivalent experience and 8+ years of post-bachelor's machine learning experience, with expertise in generative modeling, computer vision, deep learning, and ML frameworks.
TensorFlow, PyTorch, JAX, MLX, scikit-learn