Apply on Employer Site

TikTok · 2 months ago

Senior Software Engineer, Computer Vision (Knowledge Distillation)

San Jose, CA

Full-time

Onsite

Senior Level

$137K/yr - $360K/yr

5+ years exp

TikTok is the leading destination for short-form mobile video, inspiring creativity and bringing joy. They are seeking a Senior Software Engineer in Computer Vision to develop algorithm acceleration technologies and lead generative model innovation for GenAI tasks, enhancing user experience across TikTok products.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo

H1B Sponsor Likely

Responsibilities

Develop algorithm acceleration technologies for text-to-image/text-to-video models through knowledge distillation, model architecture redesign (dynamic MoE routing/sparse attention), and parameter-efficient design (low-bit quantization) to achieve order-of-magnitude efficiency gains

Lead generative model innovation with focus on diffusion acceleration (sampling step reduction/latent optimization), autoregression model efficiency

Qualification

C++PythonDiffusion modelsModel compressionDeep learning frameworksCommunication skillsTeamwork skills

Required

B.S./M.S./Ph.D. in Computer Science or related fields with 5+ years of software development experience in C++/Python

Expertise in diffusion models (Stable Diffusion/DiT) with deep understanding of computational bottlenecks and optimization methodologies

Proven experience in ≥1 domain: Image/Video Generation, Image/Video Editing/Understanding, Model Compression (Quantization/Knowledge Distillation), Efficient Architectures (MoE/Sparse attention), Generative Alignment (RLHF/DPO)

Excellent communication and teamwork skills, capable of thriving in a fast-paced work environment

Preferred

Kaggle competition achievements, publications at ICML/NeurIPS/CVPR, or open-source contributions (e.g., HuggingFace Diffusers optimization)

Research experience in GenAI /MLsys areas

Familiarity with open source deep learning frameworks such as Pytorch/DeepSpeed/Jax etc

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

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)