TikTok
Posted 1mo ago

Machine Learning Engineer, NLP - TikTok E-commerce Knowledge Graph

TikTok
San Jose, California, United States
$150k-$317k/yrOnsiteFull Time
Responsibilities
  • constructing graphs
  • developing models
  • collaborating cross-functionally
Requirements
  • Bachelor's in CS,3+ years in ML/NLP/CV,proficiency in C++/Python/Go/Java,experience with deep learning frameworks and large-scale text processing,good teamwork and communication
Technical tools mentioned
C++PythonGoJavaTensorFlowPyTorchHadoopSparkHiveFlinkLDAWord2VecELMoGPTBERT

Job description

Our team is responsible for developing state-of-the-art NLP/ML algorithms and strategies to improve user consumption experience, inspire merchants' service quality and revenue, and build a fair and flourishing ecosystem on our E-commerce Platform. More specifically, our team is responsible for the algorithms of Product Knowledge Graphs under TikTok's global e-commerce business.

What you will do:
• Participate in the development of massive knowledge graphs of real-world products to support feed ranking, recommendations, and ads.
• Collaborate with product managers, data scientists, and the product strategy & operation team to define product strategies and features.

Responsibilities:
• Knowledge graph construction, including product/content/feedback understanding and category/brand/SPU construction.
• Construct knowledge graphs of buyers and products.

Minimum Qualifications:
• Bachelor's degree in Computer Science or related technical field
• 3+ working experience in one of the following fields: machine learning, NLP, and computer vision
• Experience with software development in at least one of the following programming languages: C++, Python, Go, Java
• Good sense of teamwork and communication skills, practical experience in relevant business scenarios is preferred.

Preferred Qualifications:
• Proficient in using at least one mainstream deep learning frameworks such as TensorFlow/PyTorch, understanding distributed training, distillation acceleration, and other implementation methods.
• Experience in text classification, text matching, sequence labeling, knowledge graph.
• Aware of certain processing methods and optimization experience on domain adaptation, small sample construction, text mining, unsupervised/semi-supervised and other similar issues.
• Familiar with commonly used machine learning and deep learning algorithms, understand basic network model structure (DNN/LSTM/CNN, etc.) and text representation methods (LDA/Word2Vec/ELMo/GPT/BERT, etc.), have practical experience in deep learning training and reasoning model tuning.
• Experience in large-scale text data processing or cleaning (Such as using Hadoop/Spark/Hive/Flink).

The base salary range for this position in the selected city is $150000 - $316800 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:

Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act. Our company believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment:

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; and

3. Exercising sound judgment.

32

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 San Jose, California
1h
Save
Mark Applied
Hide
Machine Learning Engineer, Tapestry
Mountain View or United States or United Kingdom or Chile or New Zealand or Australia or Brazil
$166k-$244k/yr HybridFull Time
X, The Moonshot Factory
X, The Moonshot FactoryNASDAQ: GOOGL: Develops early-stage technologies and experimental projects.
Bachelor's or master's degree in ML, computer science, statistics, or related field; ML engineering experience; Python and PyTorch or TensorFlow; expertise in specified ML domains.
Python, PyTorch, TensorFlow, AWS, GCP, Azure
4h
Save
Mark Applied
Hide
Senior Machine Learning Engineer, Infrastructure
New York City or San Francisco
$212k-$318k/yr HybridFull Time
Patreon
Patreon: Membership platform for creators to monetize work and engage fans.
Deep experience building production ML infrastructure, low-latency inference pipelines, feature stores, distributed systems, backend engineering, Python, debugging, documentation, and code reviews.
Python
22h
Save
Mark Applied
Hide
2027 - Full Time Assistant Vice President - Corporate Functions, Analytics Lab (Machine Learning Engineer)
Jersey City or Montreal or Paris or Lisbon or London or New York or Chesterbrook or San Francisco or Boston or Chicago or Denver or Miami or Washington
$150k/yr OnsiteFull Time
BNP Paribas
BNP ParibasEuronext Paris: BNP: Provides global banking, financial, and investment services.
Master’s degree in a quantitative field; December 2026–June 2027 graduation; Python or C, SQL, statistical modeling, data analysis, LLM/agentic workflows, ML frameworks, RAG systems, Git, and communication skills.
Python, C, SQL, LangChain, CrewAI, Google ADK, OpenAI Agent SDK, TensorFlow, PyTorch, scikit-learn, Git, DVC, MLflow, Jenkins, Docker
1d
Save
Mark Applied
Hide
Senior Principal Machine Learning Engineer
San Jose, California, United States
$220k-$300k/yr OnsiteFull Time
SambaNova Systems
SambaNova Systems: Develops custom AI hardware and software for enterprise computing.
8+ YOEBachelor's degree in computer science, electrical engineering, or related field; 8+ years of ML engineering experience; deep LLM expertise; technical leadership and complex project delivery experience.
Large Language Models (LLMs), RDU, SambaStack, SambaCloud, speculative decoding, reinforcement learning, mixture-of-experts
2d
Save
Mark Applied
Hide
Founding Engineer - Machine Learning
Mountain View, California, United States
$220k-$300k/yr OnsiteFull Time
Clera
Clera: AI talent agent matching professionals with high-growth startup roles
3+ YOERequires 3–10 years of ML engineering, applied science, or research engineering experience; Python and PyTorch, TensorFlow, or JAX; distributed systems, cloud ML infrastructure, MLOps, and large-scale data experience.
Python, PyTorch, TensorFlow, JAX, AWS, GCP, Azure, Weights & Biases, MLflow
3d
Save
Mark Applied
Hide
Machine Learning Engineer
San Francisco, California, United States
$170k-$300k/yr HybridFull Time
Clay
Clay: Platform for automated lead enrichment and sales workflows.
5+ YOE5+ years in machine learning engineering or ML-heavy software engineering, production ML features, strong coding, LLMs or classical ML, data-intensive systems, and product-oriented problem solving.
LLMs, Snowflake, dbt, Dagster
3d
Save
Mark Applied
Hide
Machine Learning Engineer - Satellite Capacity Optimization & Planning
Carlsbad or San Jose or San Francisco or New York City
$141k-$222k/yr OnsiteFull Time
Viasat
ViasatNASDAQ: VSAT: Provider of global satellite-based connectivity and secure communication solutions.
7+ YOERequires 7+ years in ML or optimization, optimization techniques, production cloud ML systems, ML frameworks, geospatial visualization, containerized development, SQL, RESTful APIs, and up to 10% travel.
AWS, GCP, TensorFlow, PyTorch, SQL, RESTful APIs, Airflow, Athena, BigQuery, ECS, Batch
3d
Save
Mark Applied
Hide
Lead Machine Learning Engineering, (Hybrid)
Seattle or San Jose
$198k-$250k/yr HybridFull Time
Cisco
CiscoNASDAQ: CSCO: Develops and sells networking hardware and cybersecurity software.
8+ YOESTEM degree and 8+ years relevant experience, or equivalent advanced-degree pathways; 3+ years scaling ML datasets; 5+ years programming and ML framework experience.
Python, C++, Go, PyTorch, TensorFlow, Spark, Ray, Beam, Large Language Models (LLMs)