Software Engineer - Model Infrastructure, TikTok Feeds

TikTok
San Jose
Workplace: OnsiteFull timeFunction: Software EngineeringEducation: bachelorsSkills: ["Initiative","Self-drive","Collaboration","Independent work","Fast-paced execution"]

Build and optimize the infrastructure powering TikTok’s For You Feed recommendation system, spanning online inference and offline training. Work on large-scale recommendation model architecture and engineering, addressing computation and storage bottlenecks while improving efficiency, scalability, and stability. Collaborate with algorithm researchers to co-design infrastructure using cutting-edge ML frameworks, GPU performance tooling, and hardware accelerators to balance compute cost with recommendation effectiveness at scale.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
TikTok
TikTok
1 month ago

Software Engineer - Model Infrastructure, TikTok Feeds

✓ Verified Job

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

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

Job Summary

Build and optimize the infrastructure powering TikTok’s For You Feed recommendation system, spanning online inference and offline training. Work on large-scale recommendation model architecture and engineering, addressing computation and storage bottlenecks while improving efficiency, scalability, and stability. Collaborate with algorithm researchers to co-design infrastructure using cutting-edge ML frameworks, GPU performance tooling, and hardware accelerators to balance compute cost with recommendation effectiveness at scale.
Location: San Jose
Workplace: Onsite
Employment Type: Full time
Job Function: Software Engineering

Key Responsibilities

  • •Build and optimize infrastructure for online inference and offline training of recommendation models.
  • •Solve challenges from high model complexity, massive data, and large-scale deployments.
  • •Partner with algorithm researchers to co-design infrastructure aligned with ML frameworks and hardware accelerators.
  • •Continuously improve model architecture, efficiency, and scalability to enhance user experiences.
  • •Abstract and build reusable infrastructure components and tools for the broader recommendation stack.

Key Requirements

  • •Bachelor’s degree or above in Computer Science or related fields.
  • •Understanding of GPU architecture and the software stack (CUDA, CUTLASS, Triton Lang) with experience in GPU performance analysis.
  • •Experience with at least one machine learning framework such as TensorFlow or PyTorch (or in-house alternatives).
  • •Initiative and self-drive to take on complex challenges.
  • •Proven ability to work independently and collaboratively in a fast-paced team environment.
Education:Bachelor's in Computer Science
Skills:InitiativeSelf-driveCollaborationIndependent workFast-paced execution
Tech Stack:CUDACUTLASSTriton LangTensorFlowPyTorchMLIRXLATVM

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