Software Engineer, Machine Learning Infrastructure

TikTok
San Jose
Workplace: OnsiteFull timeFunction: Software EngineeringEducation: bachelorsSkills: ["Research","Optimization","Collaboration","Problem-solving"]

Build and optimize architecture for TikTok’s recommendation system, enabling stable, high-quality user experiences. Work on research and optimization of large-scale machine learning online prediction frameworks, process large-scale feature samples, and support data flow conversion, model training, and parameter tuning. Collaborate with algorithm teams to analyze and solve performance and scalability challenges, aligning engineering breakthroughs with practical product needs.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 month ago

Software Engineer, Machine Learning Infrastructure

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 18 hours agoStatus: Live

Job Summary

Build and optimize architecture for TikTok’s recommendation system, enabling stable, high-quality user experiences. Work on research and optimization of large-scale machine learning online prediction frameworks, process large-scale feature samples, and support data flow conversion, model training, and parameter tuning. Collaborate with algorithm teams to analyze and solve performance and scalability challenges, aligning engineering breakthroughs with practical product needs.
Location: San Jose
Workplace: Onsite
Employment Type: Full time
Job Function: Software Engineering
Seniority: Mid level

Key Responsibilities

  • •Participate in research and optimization of large-scale machine learning online prediction frameworks to support efficient recommendations.
  • •Process large-scale feature samples, including data flow conversion, model training, and parameter tuning.
  • •Address related issues through research and technical breakthroughs.
  • •Collaborate with algorithm teams to analyze and solve recommendation system challenges.
  • •Assist with practical business needs by applying infrastructure and model optimization work.

Key Requirements

  • •Bachelor’s degree or above in Computer Science or related fields, with experience building scalable systems.
  • •Proficient in C/C++ programming in a Linux environment.
  • •Familiar with operating system concepts and computer architecture.
  • •Understand underlying frameworks of mainstream machine learning frameworks (TensorFlow, PyTorch).
  • •Background in high performance computing, applied ML models (MLP/Transformer/MoE), ML hardware acceleration (e.g., GPU/RDMA), or ML systems.
Education:Bachelor's in Computer Science or related fields
Skills:ResearchOptimizationCollaborationProblem-solving
Tech Stack:C/C++LinuxTensorFlowPyTorchGPURDMA

Company Brief

TikTok
Short-form video platform that lets users create, share, and discover entertainment content through algorithmic recommendations. It also offers advertising and creator tools for brands, influencers, and businesses.
Industry: Digital Media
Company Size: Enterprise (1,001+ employees)
Growth: Scaleup
Headquarters: Singapore, Singapore
Founded: 2016
WebsiteLinkedIn