Apply on Employer Site

TikTok · 1 month ago

Machine Learning Engineer - Trae USDS

San Jose, CA

Full-time

Hybrid

Mid Level

$137K/yr - $360K/yr

3+ years exp

TikTok is the leading destination for short-form mobile video. They are seeking a Machine Learning Engineer to join the core R&D team of TRAE, responsible for model training, optimization, quantization, and deployment.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo

H1B Sponsor Likely

Responsibilities

Design, train, and fine-tune large language models (LLMs) that support TRAE’s core reasoning and code generation capabilities

Build efficient, stable, and scalable model training and evaluation pipelines

Collaborate with infrastructure and product teams to deploy and monitor models efficiently on GPU clusters

Continuously optimize models for latency, throughput, and accuracy

Stay up to date with and apply cutting-edge techniques in large model optimization and inference acceleration

Qualification

Machine LearningDeep LearningPyTorchTensorFlowPythonC++/CUDAModel OptimizationDistributed TrainingCollaborationProblem Solving

Required

Bachelor's degree or above in Computer Science, Electrical Engineering, Mathematics, or a related field

Solid foundation in machine learning, deep learning, and optimization algorithms

Proficiency in PyTorch or TensorFlow, with programming skills in Python and C++/CUDA

Experience with large-scale distributed training, mixed-precision training, or model parallelism

Hands-on experience with model quantization, pruning, or CUDA-based deployment optimization

Passionate about building efficient, production-ready AI systems and advancing the future of AI-driven software development

Preferred

Master's degree or above in Computer Science, Electrical Engineering, Mathematics, or a related field

3 years of experience in top tech companies

Benefits

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

10 paid holidays per year

10 paid sick days per year

17 days of Paid Personal Time (prorated upon hire with increasing accruals by tenure)

Company

TikTok

Glassdoor3.4

TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.

Founded in 2003

Los Angeles, California, USA

10001+ employees

https://www.tiktok.com

H1B Sponsorship

TikTok has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (979)

2024 (601)

2023 (387)

2022 (322)

2021 (133)

2020 (72)