Machine Learning Engineer Graduate (E-Commerce Supply Chain & Logistics - LLM/Agent) - 2027 Sta[...]

TikTok; San Jose (CA); On-site; USD 162,000 - 388,000; Full time

Job summary

TikTok is seeking a Machine Learning Engineer Graduate to contribute to AI-native capabilities in global e-commerce supply chain and logistics. You will work on LLMs, agents, and reinforcement learning applications at scale, shaping workflows across markets.

Qualified graduates with a PhD and strong Python plus ML framework experience will join a fast-moving, highly collaborative team with opportunities to pursue bold ideas and impactful projects in a dynamic tech company.

Qualifications

• Individuals who are completing or have recently completed a PhD in AI, CS, ML, NLP or related field.
• Experience with LLMs, agents, RAG, tool use, evaluation, or applied NLP systems.
• Strong programming in Python and at least one production language (Java/C++/Go/TypeScript).
• Familiarity with ML frameworks such as PyTorch, TensorFlow, JAX, LangChain or HuggingFace.
• Strong problem decomposition, data analysis, and communication skills.

Responsibilities

• Develop large language models, agent systems, and related intelligent systems for global e-commerce logistics.
• Build domain LLM capabilities for e-commerce supply chain including RL, CPT, SFT, optimization and inference cost.
• Create multimodal understanding for logistics data to support simulation and prediction.
• Design core agent architecture, memory/state management, and tooling routing for safe execution.

Skills

PhD in AI/CS

LLMs/Agents

Python

Java/C++/Go/TS

ML frameworks

RAG & tools

Problem solving

Education

PhD in AI/CS/ML

Tools

PyTorch

TensorFlow

JAX

HuggingFace

LangChain

LlamaIndex

vLLM

Job description

Discover a career that energizes and excites you every day.

@2026 TikTok

Technology

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

Location:

San Jose

Employment Type:

Regular

Job Code:

A104293A

Share this listing:

Responsibilities

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 our team. As a graduate, you will get opportunities to pursue bold ideas, tackle complex challenges, and unlock limitless growth.Successful candidates must be able to commit to an onboarding date by the end of the year. Please state your availability and graduation date clearly in your resume.

• Responsible for the development of large language models, agent systems, and related intelligent systems for the supply chain and logistics of the global E-Commerce business.
• Build and land domain LLM model capabilities for e-commerce supply chain and logistics, covering 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 business applications such as address correction, trajectory prediction, logistics cost analysis, customer service semantic understanding, and root-cause analysis.
• Develop 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.
• Build core agent capabilities for team and business workflows, including AutoResearch for new solution exploration, 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.
• Design and improve agent architecture and engineering systems, 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 evolution mechanisms that improve context, skills, workflows, and model behavior over time.

Qualifications

Minimum Qualifications

• Individuals who are completing or have recently completed a PhD degree in in artificial intelligence, computer science, machine learning, natural language processing, data mining, software engineering or a related discipline.
• Experience with LLMs, agents, RAG, tool use, post-training, evaluation, or applied NLP systems, with the ability to connect model behavior to business and engineering requirements.
• Strong programming ability in Python and at least one production-oriented language such as Java, C++, Go, or TypeScript; comfortable building end-to-end systems, not only model experiments.
• Familiarity with machine learning and deep learning frameworks such as PyTorch, TensorFlow, JAX, vLLM, Hugging Face, LangChain, LlamaIndex, or similar ecosystems.
• Strong problem decomposition skills, data analysis capability, and communication skills; able to work with ambiguous business questions and convert them into executable technical plans.

Preferred Qualifications

• Hands‑on experience with coding agents, long‑horizon agents, computer‑use agents, agent harnesses, workflow orchestration, or automated evaluation frameworks.
• Experience with LLM post‑training, including SFT, DPO, PPO, GRPO, RLHF, RLAIF, reward modeling, counterfactual data, or evidence‑driven decision training.
• Experience building benchmarks from real production questions, including taxonomy design, golden answer construction, error attribution, and regression evaluation.
• Experience with logistics, e‑commerce, operations research, data platforms, knowledge graphs, or enterprise knowledge management systems.
• Published papers or strong open‑source work in LLMs, agents, NLP, data mining, machine learning systems, evaluation, or AI engineering.

Job Information

The base salary range for this position in the selected city is $162000 - $387600 annually.

Compensation may vary outside of this range depending on a number of factors, including a candidate’s qualifications, skills, competencies and experience, and location. Base pay is one part of the Total Package that is provided to compensate and recognize employees for their work, and this role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.

Benefits may vary depending on the nature of employment and the country work location. Employees have day one access to medical, dental, and vision insurance, a 401(k) savings plan with company match, paid parental leave, short‑term and long‑term disability coverage, life insurance, wellbeing benefits, among others. Employees also receive 10 paid holidays per year, 10 paid sick days per year and 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).

The Company reserves the right to modify or change these benefits programs at any time, with or without notice.

For Los Angeles County (unincorporated) Candidates:

1. Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;

2. Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems;

3. Exercising sound judgment.

About TikTok

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.

Why Join Us

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.

TikTok Accommodation

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