TikTok · 9 hours ago
Sr. Multimodal Model Training and Inference Optimization Engineer
San Jose, CA
Full-time
Onsite
Mid, Senior Level
$209K/yr - $438K/yr
3+ years expTikTok is the leading destination for short-form mobile video, and they are seeking an experienced Multimodal Model Training and Inference Optimization Engineer. The role involves optimizing AI model training and inference, enhancing the performance and scalability of large-scale generative AI models.
Social MediaVideoMedia and EntertainmentMarketingContent CreatorsContent Discovery
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
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 etc
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
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
TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.
10001+ employees
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 StageLeadership Team
N Ali Mohamed
CEO
Blake Chandlee
VP Global Business Solutions
Recent News
Latest Nigeria News | Top Stories from Ripples Nigeria
2026-02-09
Mobile World Live
2026-02-09
Company data provided by crunchbase