Machine Learning Engineer, E-commerce Recommendation Foundation - USDS

Job summary

San Jose
Software Developer

Work model

Fully remote
2 days ago
Job description

Responsibilities

The E-commerce Recommendation Foundation team is dedicated to building the next-generation recommendation intelligence. We aim to develop a unified Foundation Model that supports multi-business and multi-scenario recommendation systems, covering the full pipeline from retrieval and ranking to re-ranking, and driving a comprehensive upgrade in intelligence and generative capability.

We believe the future of recommendation systems goes beyond predicting click-through rates --- it lies in understanding the relationship between people and content, and in generating new connections. The team is exploring an event-sequence-driven generative recommendation paradigm, deeply integrating large language models (LLMs), multimodal understanding, reinforcement learning, and system optimization to advance recommendation systems toward general-purpose intelligent agents.

We value original exploration and encourage both research thinking and engineering excellence. Every team member is empowered to propose hypotheses and validate ideas in an open environment --- your code and papers may help define the next paradigm of recommendation systems. We seek individuals with a general intelligence mindset to join us in redefining the future of recommendation.

  • Build and optimize cross-scenario shared Foundation Models to enable unified modeling and efficient inference. Advance the event-sequence-driven generative recommendation paradigm, integrating multimodal understanding and generative capabilities.
  • Apply LLM technologies across retrieval, ranking, and re-ranking stages; participate in model training, inference optimization, and system co-design.
  • Explore the integration of LLMs / VLMs with recommendation systems to develop adaptive and evolving intelligent recommenders.
  • Research end-to-end generative recommendation and system optimization methods that balance efficiency and user experience.

Qualifications

Minimum Qualifications

  • Solid theoretical foundation in machine learning, deep learning, or information retrieval.
  • Proficiency in Python and familiarity with mainstream deep learning frameworks (e.g., PyTorch).
  • Strong passion for intelligent recommendation systems and a self-driven research mindset.

Preferred Qualifications

  • Experience in large-scale recommendation system development or large-model training, with notable technical achievements in a sub-area.
  • Research experience or publications in LLMs, multimodal learning, reinforcement learning, or generative recommendation.
  • Familiarity with pre-training and post-training processes for large language models (LLMs) or Foundation Models.

About USDS

TikTok USDS Joint Venture LLC is dedicated to the safety and security of millions of Americans who create, discover, and connect with what they love on the apps we operate. The Joint Venture has been established in compliance with the Executive Order signed by President Trump on September 25, 2025. Our foundation is a comprehensive data privacy and cybersecurity program we operate under defined safeguards to protect national security and secure U.S. user data, apps and the algorithm.

On-site presence across teams allows the company to operate with greater speed, alignment, and agility --- especially in areas like real-time decision-making, team development, and integrated execution. As such, 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. Our innovative product is built to help people authentically express themselves, discover and connect -- and our global, diverse teams make that possible.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives.

USDS Reasonable Accommodation

USDS 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/USDS-RA

Job Information

Compensation Description (Annually)

The base salary range for this position in the selected city is $136,800 - $359,720 annually. Compensation may vary depending on a number of factors, including a candidate's qualifications, skills, competencies and experience, and location. This role may be eligible for additional discretionary bonuses/incentives, and restricted stock units.

Benefits

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.