Machine Learning Engineer Graduate (Conversational AI) - 2027 Start

TikTok Inc
  • Seattle, WA
    6 days ago

    Job Description

    Team Introduction: We build the next-generation unified Agent system for TikTok s global e-Commerce customer service - running in 30+ languages across one of the largest e-Commerce surfaces on the internet.

    Our north star is a self-evolving Agent: post-training, harness, memory / context engineering, tools, and evaluation form one closed loop, and every served conversation becomes the next iteration s training / eval / retrieval / skill-induction signal. This loop is already running in production - cases are mined, root-caused, turned into constrained candidates, replayed against frozen regression sets, and shipped behind guardrails.

    Two things make this team different from most "LLM application" work:

    • We build the agent runtime itself - Codex / Claude-Code-class - not prompts on top of a vendor API.
    • Evaluation and experimentation are first-class systems, not an afterthought. A self-improving loop optimizes whatever signal you give it, so the hardest and most valuable engineering here is making the judgment trustworthy - not just making the model change.

    By combining generative recommendation, large recommendation models, multimodal representation learning, and cross-domain value modeling, the team works on some of the most important algorithmic problems in live commerce. Our goal is to improve user experience, optimize ecosystem efficiency, and drive sustainable business growth for TikTok Shop across global markets.

    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. Candidates can apply to a maximum of two positions and will be considered for jobs in the order you apply. The application limit is applicable to our Company and its affiliates jobs globally. Applications will be reviewed on a rolling basis - we encourage you to apply early.

    Responsibilities:

    • Agent runtime (harness / agent loop). Orchestrate skills, tools, and context; implement loop control & intervention, progressive disclosure, and behavior-level guardrails. Build the production safety layer - pre-flight budgets and timeout truncation, serve-time gates, shadow / swap-in answer delivery, and safe fallback paths.
    • Context & memory for long multi-turn agents. Agentic memory (structured note-taking), context compaction / summarization, context editing / observation masking, and just-in-time (retrieve-then-load) retrieval. Treat context as an evolving, itemized playbook - with structured diffs and a deterministic curator - rather than an ever-growing prompt.
    • Post-training & the data flywheel. SFT / DPO / RL to internalize rules into weights (so the prompt gets shorter, not longer), plus distillation to smaller serving models. Turn served conversations into training / eval / retrieval signals.
    • Tools, Skills, and MCP. Tools-as-APIs, connectors, skill / tool search for large inventories, and skill-library governance - description conflicts, trigger evals, cross-skill mis-fire matrices, and on-demand loading instead of dumping every definition into context.
    • Evaluation you can bet a launch on. LLM-as-judge with human-agreement calibration; statistical rigor - paired comparison, confidence intervals, repeated sampling, pass^k; held-out and time-rolling eval splits with overfitting alarms; cascaded scoring and cross-family judge panels to make evaluation affordable at scale.
    • The self-evolving loop. Case mining automatic root-cause constrained candidate generation replay verification against frozen regression sets canary flywheel. Build the plumbing that makes it auditable: candidate registry with exact runtime read-back, change lineage, and an archive of rejected candidates you can sample from next round.
    • Online experimentation & causal readout. Shadow / canary / A-B, non-inferiority gates, traffic-split health, metric definitions that survive scrutiny, and off-policy counterfactual evaluation where live A/B isn t possible. Minimum Qualification(s):
    • Individuals who are completing or have recently completed a Bachelor's degree in Computer Science, Electrical Engineering, Mathematics, Statistics or a related discipline
    • Strong Python plus one of C++ / Go / Rust / Java
    • Solid ML / DL / NLP fundamentals, with genuine hands-on experience with LLMs or agents (coursework, research, internship, competition, open-source, or a serious side project)
    • Basic statistical literacy - you can compute a confidence interval, explain what a p-value does and doesn t mean, and tell the difference between "the number went up" and "the system got better"
    • Able to read a paper or an engineering blog and turn it into working code

    Preferred Qualification(s)

    • Have built the runtime, not just called an API - even at research / hobby / competition scale: your own agent loop / harness, a memory / context-management system, a RAG or tool-use agent, or a fine-tuned / post-trained model
    • Hands-on experience in any one of the following - depth in one is enough, breadth welcome:
    • Post-training: SFT / DPO / RLHF / RLAIF / RLVR, reward modeling, reward hacking and how to defend against it
    • Agent systems: harness, context engineering, MCP / Skills, sub-agents, tool search
    • Evaluation & experimentation: LLM-as-judge and judge calibration, pass^k, regression suites, A/B and non-inferiority testing, off-policy evaluation
    • Self-improving / evolutionary systems: evolutionary program search, candidate archives and parent sampling, automatic prompt / context optimization, multi-objective (Pareto) selection and credit assignment
    • Inference & serving: vLLM / TensorRT-LLM, MoE, KV / prompt caching, and the cost engineering that comes with it
    • Multilingual NLP
    • Publications (for PhD), strong competition results (ACM-ICPC / Kaggle / ML competitions), or notable open-source contributions
    • Taste for the honest number. We d rather hire someone who says "this result isn t significant yet" than someone who ships a green dashboard. If you ve ever killed your own experiment because the evidence didn t hold up, tell us about it.
    • e-Commerce or multilingual experience is a plus, not required

    Numbers & Facts

    LocationSeattle, WA

    Skills

    • Algorithmsunmatched
    • Application Programming Interface (API)unmatched
    • Artificial Intelligence (AI)unmatched
    • Blogunmatched
    • Budgetingunmatched
    • Business Growthunmatched
    • C++ Programming Languageunmatched
    • Cachingunmatched
    • Calibrationunmatched
    • Candidate Sourcingunmatched
    • Computer Hackingunmatched
    • Computer Scienceunmatched
    • Cost Engineeringunmatched
    • Customer Support/Serviceunmatched
    • Data Scienceunmatched
    • Ecosystemsunmatched
    • Editingunmatched
    • Electrical Engineeringunmatched
    • Go Programming Language (Golang)unmatched
    • Green Businessunmatched
    • Javaunmatched
    • Just in Time (JIT)unmatched
    • Literacyunmatched
    • MCP - Microsoft Certified Professionalunmatched
    • Machine Learningunmatched
    • Management of Information Systems/Technology (MIS)unmatched
    • Mathematicsunmatched
    • Memory Hardwareunmatched
    • Metricsunmatched
    • Multilingualunmatched
    • Natural Language Processing (NLP)unmatched
    • Onboardingunmatched
    • Open Sourceunmatched
    • Plumbingunmatched
    • Policy Evaluationunmatched
    • Publicationsunmatched
    • Python Programming/Scripting Languageunmatched
    • Reporting Dashboardsunmatched
    • Rust Programming Languageunmatched
    • Search Agentunmatched
    • Statisticsunmatched
    • Testingunmatched
    • User Interface/Experience (UI/UX)unmatched
    • eCommerceunmatched

    Be found by employers

    5,500+ employers search our resume database daily. Add yours to get found by recruiters looking for candidates like you.

    Level up your application

    Professional resume templates

    Browse dozens of recruiter approved resume templates, layouts and formats. Choose your favorite and make it your own in minutes.

    Free resume templates

    Resume Builder

    Improve your existing resume or start from scratch and create a standout, ATS-friendly resume. Add job-specific content, download and apply.

    Resume Builder