T

TikTok

Short-form video platform with ads

Software Engineer - Machine Learning Infrastructure

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSan Jose, CA, USAIn-person

About the Job

Requirements

  • A bachelor's degree or above, majoring in Computer Science or a related field, and experience building scalable systems.
  • Proficiency in C and C++ programming in a Linux environment.
  • Familiarity with operating system concepts and computer architecture.
  • Understanding of the underlying frameworks of mainstream machine learning frameworks such as TensorFlow and PyTorch.
  • A background in at least one of the following fields: High Performance Computing, Applied ML models (MLP/Transformer/MoE), ML Hardware Acceleration (such as GPU/RDMA), or ML Systems.

Responsibilities

  • Participate in research and optimization of large-scale machine learning online prediction frameworks to support efficient model recommendation for TikTok products.
  • Process large-scale feature samples, including data flow conversion, model training, and parameter tuning, and address related issues through research and breakthroughs.
  • Collaborate closely with algorithm teams to analyze and solve these challenges and assist with practical business needs.

Desired Qualifications

  • Experience in deep learning model training and prediction.
  • Experience in algorithm implementation or optimization of existing algorithms.
  • Familiarity with model pruning, quantization, and compression methods.
  • Experience in software and hardware integration and optimization.

About the Company

TikTok

Short-form video platform with ads

TikTok is a short-form mobile video platform that lets people create, discover, and share quick videos. Users upload vertical clips, edit them with built-in tools, and browse a personalized feed powered by an algorithm that highlights relevant content. Advertisers can use in-feed ads, branded hashtags, and the Business Center to plan and measure campaigns, making it easy for brands to reach a global audience. Its goal is to inspire creativity and bring joy by helping people express themselves and by connecting brands with a wide audience.

Company size
10,001+
Company stage
Grant
Total funding
$740K
Headquarters
Santa Monica, CA
Founded
2016

Recently Posted Jobs

Explore more opportunities at TikTok

See more jobs