884 jobs for "Lead AI Engineer"

Software Engineer Project Intern (Model Infrastructure) - 2026 Start (BS/MS)

At TikTok - San Jose

Posted on Apr 4

View job Apply on company site

Machine Learning Engineer - TikTok BRIC Live Integrity - Singapore

At TikTok

Posted on Feb 9

View job Apply on company site

Senior Machine Learning Engineer - TikTok Short Video Content Understanding/Multimodal Recommendation

At TikTok - San Jose

Posted on Sep 24

View job Apply on company site

Machine Learning Engineer - TikTok Short Video Content Understanding/Multimodal Recommendation

At TikTok - San Jose

Posted on Aug 6

View job Apply on company site

Problem Management Director

At DTCC - Coppell

Posted on Nov 7

View job Apply on company site

Vice President, Client Enrollments

At Achieve - Tempe · Orlando

Posted on Nov 4

View job Apply on company site

Product Manager Associate Director - Technology Research & Innovation

At DTCC - Jersey City

Posted on Oct 27

View job Apply on company site

.NET Application and Solutions Architect - Assistant Vice President

At iCapital - Salt Lake City

Posted on Oct 23

View job Apply on company site

.NET Application and Solutions Architect - Assistant Vice President

At iCapital - Salt Lake City

Posted on Oct 22

View job Apply on company site

Sr. Product Engineer - National Imagery

At Esri - Remote / Flexible

Posted on Oct 15

View job Apply on company site

Sr. Product Engineer - National Imagery

At Esri - Vienna

Posted on Sep 4

View job Apply on company site

Senior Technical Product Manager - Modernization

At Appian - Chennai

Posted on Sep 3

View job Apply on company site

.NET Application and Solutions Architect - Assistant Vice President

At iCapital - Toronto

Posted on Jul 24

View job Apply on company site

.NET Application and Solutions Architect - Assistant Vice President

At iCapital - Toronto

Posted on Jul 23

View job Apply on company site

Chief Engineer - Critical Power Solutions

At Mitsubishi Electric Power Products - Warrendale

Posted on May 15

View job Apply on company site

Senior Engineer of Audio and Visual Operation - IT

At TikTok - London

Posted on Nov 2

View job Apply on company site

Machine Learning Engineer - Emerging Business - TikTok BRIC

At TikTok

Posted on Oct 29

View job Apply on company site

Lead SRE

At Apple

Posted on Oct 19

View job Apply on company site

AIML - Machine Learning Engineer, Visual Intelligence Technology

At Apple - Santa Clara

Posted on Oct 17

View job Apply on company site

AIML - Machine Learning Engineer, Visual Intelligence Technology

At Apple - Santa Clara

Posted on Oct 17

View job Apply on company site

TikTok

Software Engineer Project Intern (Model Infrastructure) - 2026 Start (BS/MS)

San JosePosted Apr 4, 2026

Apply on company site View full job

Why you should apply for a job to TikTok:

4.5/5 in overall job satisfaction
4.5/5 in supportive management
100% say women are treated fairly and equally to men
100% would recommend this company to other women
100% say the CEO supports gender diversity

Ratings are based on anonymous reviews by Fairygodboss members. See all reviews ›

Responsibilities

About the Team
The TikTok Model Infrastructure team is the core engine powering the world's most engaged "For You" feed. We focus on the engineering efficiency and architectural evolution of recommendation models at an unprecedented scale. As we lead the industry's shift toward LLM2Rec and Large Recommendation Models (LRM), our mission is to build ultra-high-performance infrastructure that bridges the gap between massive data scale and extreme algorithmic complexity.
We tackle the industry's most demanding "frontier" challenges: managing Petabyte-scale distributed embedding states, optimizing thousand-node GPU clusters, and perfecting real-time Sparse/Dense streaming. Our work ensures that models with hundreds of billions of dense parameters-on par with the world's largest LLMs-can operate with millisecond-level latency.

We are seeking Software Engineering Interns to join the Model Infra team to redefine the performance boundaries of recommendation systems. In this role, you will focus on the efficiency of the entire model lifecycle. You will work on the convergence of generative AI and recommendation architecture, optimizing everything from the raw throughput of multi-billion parameter dense blocks to the efficient retrieval of sparse features across massive distributed memory fabrics.

As a project intern, you will have the opportunity to engage in impactful short-term projects that provide you with a glimpse of professional real-world experience. You will gain practical skills through on-the-job learning in a fast-paced work environment and develop a deeper understanding of your career interests.

Applications will be reviewed on a rolling basis - we encourage you to apply early.

Responsibilities
- Engineering Efficiency at Scale: Drive the optimization of training and inference pipelines to maximize hardware utilization (MFU/HFU) for models featuring hundreds of billions of dense parameters.
- LLM2Rec Infrastructure: Architect specialized systems to support the integration of LLMs into the recommendation stack, focusing on memory-efficient attention mechanisms and advanced KV cache management for long-sequence user modeling.
- Massive Sparse & Dense Streaming: Build and optimize high-concurrency engines for Petabyte-scale streaming training, handling continuous parameter updates and high-frequency data ingestion without compromising stability.
- Hardware-Aware Co-Design: Work closely with researchers to design next-generation recommendation architectures optimized for modern GPU/NPU interconnects, ensuring high-bandwidth utilization across the cluster.
- Distributed State Management: Innovate on how we store and synchronize massive model states across heterogeneous memory hierarchies (HBM, DDR, and NVMe).

Qualifications

Minimum Qualification(s)
- Currently pursuing an Undergraduate/Master in Software Development, Computer Science, Computer Engineering, or a related technical discipline.
- Strong programming skills in C++ and Python.
- Solid understanding of Computer Architecture and the GPU software stack (CUDA, Triton, or NCCL).
- Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and a desire to "look under the hood" of model execution runtimes.
- A strong interest in solving system-level bottlenecks in large-scale distributed environments.

Preferred Qualification(s)
- Experience with Transformer-based architectures, 3D parallelism (TP/PP/DP).
- Deep understanding of the torch.compile stack, including TorchDynamo (graph acquisition) and TorchInductor (lowering).
- Hands-on experience writing high-performance kernels or optimizing collective communication (e.g., customizing NCCL/UCX).
- Familiarity with RDMA networking, high-performance storage, or specialized Parameter Server architectures.
- Success in programming competitions (ACM-ICPC) or contributions to prominent open-source AI infrastructure or high-performance computing projects.

Responsibilities

Qualifications

Qualifications

Minimum Qualification(s)
- Currently pursuing an Undergraduate/Master in Software Development, Computer Science, Computer Engineering, or a related technical discipline.
- Strong programming skills in C++ and Python.
- Solid understanding of Computer Architecture and the GPU software stack (CUDA, Triton, or NCCL).
- Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and a desire to "look under the hood" of model execution runtimes.
- A strong interest in solving system-level bottlenecks in large-scale distributed environments.

Preferred Qualification(s)
- Experience with Transformer-based architectures, 3D parallelism (TP/PP/DP).
- Deep understanding of the torch.compile stack, including TorchDynamo (graph acquisition) and TorchInductor (lowering).
- Hands-on experience writing high-performance kernels or optimizing collective communication (e.g., customizing NCCL/UCX).
- Familiarity with RDMA networking, high-performance storage, or specialized Parameter Server architectures.
- Success in programming competitions (ACM-ICPC) or contributions to prominent open-source AI infrastructure or high-performance computing projects.

About TikTok

TikTok is the leading destination for short-form mobile video and our mission is to inspire creativity and bring joy.

Size: 10,000+ employees
Industry: Technology: Consumer Internet

View company profile