Tech Lead, Senior Software Engineer, On-Device AI

$224,000 - $410,000/Yr

Tiktok - San Jose, CA

Full-time - Senior

San Jose, CA

Computing Infrastructure Providers, Data Processing, Web Hosting, and Related Services

About the position

TikTok is the leading destination for short-form mobile video, and our mission is to inspire creativity and bring joy. The On-Device AI Team at TikTok is at the forefront of embedding advanced AI technologies directly into devices worldwide. Leveraging our proprietary AI engine, we streamline the creation, integration, testing, and deployment of AI-driven features. Our work enhances user experience by powering diverse functionalities, including visual enhancements, video editing tools, and creative camera filters, both within TikTok and other applications. As a Tech Lead and Senior Software Engineer, you will be responsible for developing and enhancing the core On-Device AI engine for diverse platforms. You will employ efficient machine learning techniques and hardware optimizations to improve algorithmic performance. Your role will also involve seamlessly integrating state-of-the-art AI algorithms into TikTok and additional products, translating emerging research into practical, user-centric applications. At TikTok, we believe that every challenge is an opportunity to learn, innovate, and grow as one team. We are committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our mission is to inspire creativity and bring joy, and we are passionate about celebrating our diverse voices and creating an environment that reflects the many communities we reach.

Responsibilities

• Develop and enhance the core On-Device AI engine for diverse platforms.
• Employ efficient machine learning techniques and hardware optimizations to improve algorithmic performance.
• Seamlessly integrate state-of-the-art AI algorithms into TikTok and additional products.
• Translate emerging research into practical, user-centric applications.

Requirements

• B.Sc/M.Sc/PhD in Computer Science or in a related field
• 5+ years of industrial Python/C++ experience, solid CS fundamentals and problem-solving skills
• Experience in one or more of the following specializations: Generative AI, Deep Learning, Computer Vision.
• Excellent communication and teamwork skills, capable of thriving in a cross-functional environment.

Nice-to-haves

• Demonstrated expertise in advanced algorithm system design and implementation, including but not limited to inference engines, algorithm pipelines, e2e ML solutions and LLM/AIGC applications.
• Deep knowledge in machine learning optimizations such as quantization, pruning, knowledge distillation, Neural Architecture Search (NAS), and optimizations for large models.
• Proficiency in enhancing performance across various hardware platforms, including CPUs, GPUs, DSPs, and NPUs.

Benefits

• 100% premium coverage for employee medical insurance
• Approximately 75% premium coverage for dependents
• Health Savings Account (HSA) with a company match
• Dental, Vision, Short/Long term Disability, Basic Life, Voluntary Life and AD&D insurance plans
• Flexible Spending Account (FSA) Options like Health Care, Limited Purpose and Dependent Care
• 10 paid holidays per year
• 17 days of Paid Personal Time Off (PPTO)
• 10 paid sick days per year
• 12 weeks of paid Parental leave
• 8 weeks of paid Supplemental Disability
• Mental and emotional health benefits through EAP and Lyra
• 401K company match
• Gym and cellphone service reimbursements