TikTok
Posted 1mo ago

Machine Learning Engineer(NLP/CV/Multimodal), TikTok E-commerce Knowledge Graph

TikTok
San Jose, California, United States
$150k-$388k/yrOnsiteFull Time
Responsibilities
  • constructing graphs
  • developing algorithms
  • collaborating teams
Requirements
  • 3+ years in machine learning/NLP/multimodal/CV
  • CS degree preferred
  • Proficiency in C++/Python/Go/Java
  • Experience with deep learning frameworks and large-scale text processing
Technical tools mentioned
C++PythonGoJavaTensorFlowPyTorchHadoopSparkHiveFlinkLDAWord2VecELMoGPTBERTDNNLSTMCNN

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(Natural Language Processing), multimodal, 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, and 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 - $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:

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.

33

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

Short-form mobile video and social media platform.

Similar jobs

Machine Learning Engineer roles near San Jose, California
4h
Save
Mark Applied
Hide
Senior Machine Learning Engineer, Multimodal Perception
Mountain View, California, United States
$213k-$263k/yr HybridFull Time
Waymo
Waymo: Autonomous driving technology and robotaxi service provider.
2+ YOE2–5+ years training and releasing ML models in autonomous driving, robotics, or spatial AI; experience across perception and planning, learned policies, PyTorch/JAX, simulation evaluation, and production releases.
PyTorch, JAX, Vision-Language-Action (VLA), World Models
7h
Save
Mark Applied
Hide
Machine Learning Engineer, Performance Tooling
London or Sunnyvale
OnsiteFull Time
Wayve
Wayve: British autonomous-driving software licensing vehicle-agnostic AI Driver technology to automakers and fleet owners.
Requires deep performance engineering, end-to-end tool or service ownership, strong Python, PyTorch model development, data analysis, profiling, optimization, and quantitative communication skills.
Python, PyTorch
1d
Save
Mark Applied
Hide
Senior Machine Learning Engineer
Fremont, California, United States
$155k-$200k/yr HybridFull Time
Velo3D
Velo3DNasdaq Capital Market: VELO: Public metal additive manufacturer providing printers, software, and engineering services to aerospace, defense, energy, and industrial customers.
Experience with machine learning pipelines for geometric data and production software engineering; strong Python or C++ skills. Manufacturing, IoT, and industrial automation knowledge preferred.
Python, C++, Flow, Assure
2d
Save
Mark Applied
Hide
Founding Machine Learning Engineer
Mountain View, California, United States
$220k-$300k/yr OnsiteFull Time
Clera
Clera: AI-powered talent agent matching candidates to startup roles.
3+ YOE3–10 years as an ML Engineer, Applied Scientist, or Research Engineer; Python and PyTorch, TensorFlow, or JAX proficiency; ML fundamentals, distributed systems, cloud ML infrastructure, and MLOps experience.
Python, PyTorch, TensorFlow, JAX, AWS, GCP, Azure, Weights & Biases, MLflow
3d
Save
Mark Applied
Hide
Machine Learning Engineer III
San Jose or Seattle
$146k-$221k/yr HybridFull Time
Expedia Group
Expedia GroupNASDAQ: EXPE: Global travel technology powering online booking platforms.
3+ YOEBachelor’s or master’s degree in a quantitative field, 3+ years of machine learning or data-driven systems experience, Python and ML framework proficiency, and experience with data pipelines and large datasets.
Python, PyTorch, TensorFlow, Spark, SQL, Databricks, AWS
3d
Save
Mark Applied
Hide
Machine Learning Engineer
San Francisco, California, United States
$260k-$300k/yr OnsiteFull Time
Hyperbound
Hyperbound: Private AI sales-coaching platform helping enterprise revenue teams practice conversations and improve performance.
Own machine learning models end to end, including training, fine-tuning, production deployment, on-device execution, evaluation frameworks, benchmarks, and regression suites.
open source models
4d
Save
Mark Applied
Hide
Machine Learning Engineer, Causal Inference, Level 5
Los Angeles or Seattle or Palo Alto or New York City or Bellevue or Santa Monica or California or Washington or New York
$209k-$313k/yr OnsiteFull Time
Snap Inc.
Snap Inc.NYSE: SNAP: Technology focused on augmented reality and communication.
5+ YOEBachelor’s degree or equivalent experience plus 5+ years in machine learning, or equivalent master’s/PhD pathways; causal inference, experimentation, production modeling, and Python expertise required.
Python, pandas, NumPy, scikit-learn, CausalM, CausalML, EconML, DoWhy
4d
Save
Mark Applied
Hide
Machine Learning Engineer, Causal Inference, Level 5
Los Angeles or Seattle or Palo Alto or New York City or Bellevue or Santa Monica or California or Washington or New York City
$209k-$313k/yr OnsiteFull Time
Snap Inc.
Snap Inc.NYSE: SNAP: Technology focused on augmented reality and communication.
5+ YOEBachelor’s degree or equivalent practical experience and 5+ years of post-bachelor’s machine learning experience, or equivalent master’s/PhD pathways, with causal inference and experimentation expertise.
Python, pandas, NumPy, scikit-learn, CausalML, CausalM, EconML, DoWhy