SIGN IN
Sr. Multimodal Model Training and Inference Optimization Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

TikTok · 11 hours ago

Sr. Multimodal Model Training and Inference Optimization Engineer

positionSeattle, WA
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$198K/yr - $416K/yr
date3+ years exp
TikTok is the leading destination for short-form mobile video. The company is seeking an experienced Multimodal Model Training and Inference Optimization Engineer to optimize AI model training and inference, focusing on enhancing performance, scalability, and deployment of large-scale generative AI models.
Social MediaVideoMedia and EntertainmentMarketingContent CreatorsContent Discovery
check
H1B Sponsor Likelynote

Responsibilities

Optimize large model training pipelines to improve efficiency, speed, and scalability
Develop and improve distributed training strategies such as data parallelism, model parallelism, pipeline parallelism and communication to accelerate model training
Benchmark and profile deep learning models to identify performance bottlenecks and optimize computational resources

Qualification

AI model training optimizationPythonDeep learning frameworksDistributed training techniquesC++CUDATransformers knowledgeDiffusion models knowledgeCommunication skillsTeamwork skillsProblem-solving skillsSelf-motivated

Required

M.S or PhD in Computer Science, Electrical Engineering, Artificial Intelligence, or a related field
3 years+ experience in AI model training optimization
Strong software engineering skills, including proficiency in Python, C++, and CUDA
Strong proficiency in deep learning frameworks such as PyTorch, Megatron and Deepspeed
Experience with distributed training techniques such as data parallelism, model parallelism, and pipeline parallelism
Knowledge of transformers and diffusion models

Preferred

Candidates with publications at conferences such as MLSys, NeurIPS, ICLR, or ICML are preferred
Strong communication and teamwork skills
Self-motivated and strong problem-solving skills
Ability to work collaboratively in multi-functional teams
Experienced in implementing and optimizing complex and performance-critical systems

Benefits

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

twittertwittertwitter
Glassdoor3.4
company-logo
TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.
dateFounded in 2003
location
Los Angeles, California, USA
people-size10001+ employees
web-link
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)

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
N Ali Mohamed
CEO
linkedin
leader-logo
Blake Chandlee
VP Global Business Solutions
linkedin

Recent News

Latest Nigeria News | Top Stories from Ripples Nigeria
TikTok faces potential fine as EU flags infinite scroll and autoplay risks
2026-02-09
EuroNews
From AI chatbots to shopping and streaming: Which mobile apps are the most downloaded in Europe?
2026-02-09
Mobile World Live
EC finds TikTok addictive features breach DSA rules
2026-02-09
Company data provided by crunchbase