Software Engineer - TikTok AI Search Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

TikTok · 2 days ago

Software Engineer - TikTok AI Search Infrastructure

TikTok is the leading destination for short-form mobile video, committed to inspiring creativity and bringing joy. They are seeking a Software Engineer for their AI Search Infrastructure team to build scalable and stable infrastructures that enhance the search experience for billions of users, utilizing cutting-edge ML/NLP technologies.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo
badNo H1Bnote

Responsibilities

Evolve TikTok’s AI Search multi-agent LLM engine, supporting ReAct + Tool calling, DAG-based workflows, and RAG paradigms
Deploy and optimize text/multimodal LLMs, including inference acceleration, model alignment during training, and reinforcement learning
Build scalable and reliable end-to-end serving platforms supporting multiple TikTok AI Search use cases, such as Q&A cards, in-app chatbot, and visual search, including both online and offline scenarios
Build large-scale data architecture for handling billion-level data records: offline computation, distributed system performance and scheduling optimization, as well as building high-availability, high-throughput, and low-latency online services
Design and build personalized AI search capabilities to achieve more accurate AI Overview and Q&A experience for users
Collaborate with modeling and product teams to deliver better AI search experience for TikTok users

Qualification

Machine Learning InfrastructureLarge-scale Systems DevelopmentLLM Serving/TrainingC++ DevelopmentDistributed DatabasesEffective CommunicationTeamwork SkillsOwnership Mindset

Required

Bachelor or advanced degree in Computer Science or related disciplines
At least 5 years of industry experience in one of the following areas: recommendation, search, Ads and ML Infra
Strong Computer Science background, proficient coding skills and solid understanding of algorithm & data structure
Have experience in at least one of the following areas: recommendation system, search engine, advertising engine, Agent/RAG engine, LLM serving/training
Effective communication and teamwork skills, strong ownership mindset

Preferred

Familiar with system performance optimization in Linux environment, experience with large-scale C++ system development are preferred
Familiar with large-scale distributed systems development; experience with distributed databases or distributed data processing frameworks is a plus
Experience with GPU inference optimization, LLM/VLM serving and training are preferred

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 is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.

Funding

Current Stage
Late Stage

Leadership Team

leader-logo
N Ali Mohamed
CEO
linkedin
leader-logo
Blake Chandlee
VP Global Business Solutions
linkedin
Company data provided by crunchbase