Job Board
LogoLogo

Get Jobs Tailored to Your Resume

Filtr uses AI to scan 1000+ jobs and finds postings that perfectly matches your resume

TikTok Logo

Software Engineer - Machine Learning Infrastructure - Usds

TikTok

$136,800 - $359,720
Sep 17, 2025
San Jose, CA, USA
Apply Now

The company's AML team is looking to advance its next-generation AI infrastructure and recommendation platform for various ranking systems (ads, search, live & ecom) and drive substantial impact on core businesses. This role specifically aims to support and advance this mission by focusing on Machine Learning Infrastructure.

Requirements

  • Proficient in at least one programming language such as Go/Python in Linux environment, with excellent coding skills.
  • Familiar with open source distributed scheduling/orchestration/storage frameworks, such as Kubernetes (K8S), Yarn (Flink, MapReduce), Mesos, Celery, HDFS, Redis, S3, etc., with rich practical experience in machine learning system development.
  • Experience in developing and deploying large-scale systems.
  • Experience contributing to an open sourced machine learning framework (TensorFlow/PyTorch).
  • Experience in big data frameworks (e.g., Spark/Hadoop/Flink), experience in resource management and task scheduling for large scale distributed systems.

Responsibilities

  • Responsible for the design and implementation of a global-scale machine learning system for feeds, ads and search ranking models.
  • Responsible for improving use-ability and flexibility of the machine learning infrastructure.
  • Responsible for improving the workflow of model training and serving, data pipelines, storage system and resource management for multi-tenancy machine learning systems.
  • Responsible for designing and developing key components of ML infrastructure and mentoring interns.

Other

  • In order to enhance collaboration and cross-functional partnerships, among other things, at this time, our organization follows a hybrid work schedule that requires employees to work in the office 3 days a week, or as directed by their manager/department.
  • As a condition of employment, all successful candidates must be able to establish authorization to work in the United States.
  • For this position, the Company does not provide sponsorship for any immigration-related benefits.
  • This role requires the ability to work with and support systems designed to protect sensitive data and information. As such, this role will be subject to strict national security-related screening.