Machine Learning Engineer Graduate (E-Commerce)

About the Role

This graduate Machine Learning Engineer role within the Global E-commerce team at TikTok USDS Joint Venture focuses on architecting and developing sophisticated LLM-based systems. The Global E-commerce team is dedicated to providing a tailored and efficient shopping experience through short-form video and live-streaming, while maintaining a strong governance and risk control framework. As a graduate, you will have opportunities to pursue bold ideas, tackle complex challenges, and unlock growth. Successful candidates must commit to an onboarding date by the end of the year.

Key Responsibilities

• Architect and develop sophisticated LLM-based systems to process high-dimensional unstructured data (e.g., text, logs, conversations) from diverse sources.
• Define and implement complex semantic understanding frameworks, including fine-grained intent classification and entity extraction, to achieve deep contextual awareness.
• Apply state-of-the-art techniques to enhance model performance, including Instruction Tuning, RLHF/DPO, and Contextual Prompt Engineering.
• Optimize model serving for production environments using quantization, distillation, and efficient attention mechanisms (e.g., FlashAttention, vLLM).

Requirements & Tech Stack

• Minimum Qualification(s):

• Individuals who are completing or have recently completed a Bachelor's/Master's degree in Computer Science, Engineering, Math, Statistics, or a related discipline.

• Experience in one or more of the following fields: Machine Learning, Natural Language Processing (NLP), or Computer Vision.

• Proficiency in software development with at least one of the following programming languages: C++, Python, Go, or Java.

• Preferred Qualifications:

• Experience in fine-tuning, distillation, or reinforcement learning for large language models in conversational AI applications.

• Proficiency in multilingual NLP, machine translation, and cross-lingual dialogue modeling.

• Experience in developing and optimizing e-commerce video and product multimodal large models, combining LLMs with video/product representations to support tasks like multimodal classification, video QA, and cross-modal retrieval.

• A strong understanding of e-commerce policies, dispute resolution workflows, and merchant-buyer interactions to inform AI service design.

Job Information

• Base salary range: $122,574 - $187,200 annually.
• Compensation may vary based on qualifications, skills, competencies, experience, and location.
• Role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.
• Benefits may vary depending on employment nature and work location.
• Day one access to medical, dental, and vision insurance.
• 401(k) savings plan with company match.
• Paid parental leave.
• Short-term and long-term disability coverage.
• Life insurance.
• Wellbeing benefits.
• 10 paid holidays per year.
• 10 paid sick days per year.
• 17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure).

About TikTok USDS Joint Venture

TikTok USDS Joint Venture LLC is dedicated to the safety and security of millions of Americans who create, discover, and connect with content on TikTok's apps. The Joint Venture operates under defined safeguards to protect national security and secure U.S. user data, apps, and the algorithm. They safeguard the U.S. content ecosystem, holding decision-making authority for trust and safety policies and moderation. The company is shifting from a hybrid work model to a fully in-person schedule up to 5 days a week.

Why Join Us

Inspiring creativity is at the core of TikTok's mission. The innovative product is built to help people authentically express themselves, discover, and connect. The team is committed to creating value for communities, inspiring creativity, and bringing joy. They lead with curiosity, humility, and a desire to make an impact in a rapidly growing tech company. Every challenge is an opportunity to learn and innovate as one team, embracing challenges and fostering an "Always Day 1" mindset.

Note

• State your availability and graduation date clearly in your resume.
• Candidates can apply to a maximum of two positions globally.
• Applications will be reviewed on a rolling basis.