AI/ML Infra Engineer Intern: Large-Scale Rec Systems

TikTok; San Jose (CA); On-site; USD 51,000 - 73,000; Full time

Benefits offered by this job

Health insurance

Life insurance

Wellbeing benefits

Housing allowance

10 paid holidays per year

Paid sick time

Job summary

TikTok is seeking interns to work on the Model Infrastructure team, building the next generation of AI infrastructure for For You recommendations and large language models. You will contribute to distributed training, high-performance GPU computing, and scalable LLM infrastructure while collaborating with researchers and engineers.

The internship program offers hands-on experience, industry exposure, and opportunities to apply knowledge to real-world challenges, with benefits such as health and

Qualifications

• Currently pursuing a Bachelor's or Master's degree in Artificial Intelligence, Software Development, Computer Science, Computer Engineering or a related technical discipline.
• Strong programming skills in C++ or Python.
• Good understanding of data structures, algorithms, and computer systems.
• Familiarity with PyTorch or TensorFlow.
• Knowledge of Transformer architectures and Large Language Models (LLMs).
• Strong problem-solving skills and a passion for building large-scale AI systems.

Responsibilities

• Build and optimize infrastructure for large-scale model training and online inference.
• Develop distributed systems supporting large recommendation models and LLMs.
• Improve training and inference performance through GPU optimization and efficient communication.
• Collaborate with researchers to develop and deploy LLM training and serving solutions.
• Analyze system bottlenecks and implement performance optimizations.

Skills

C++

Python

Data structures

Algorithms

PyTorch

TensorFlow

LLMs

Education

Bachelor's or Master's in AI/CS/Software/Engineering

Job description

TikTok is seeking interns to work on the Model Infrastructure team, building the next generation of AI infrastructure for For You recommendations and large language models. You will contribute to distributed training, high-performance GPU computing, and scalable LLM infrastructure while collaborating with researchers and engineers.

The internship program offers hands-on experience, industry exposure, and opportunities to apply knowledge to real-world challenges, with benefits such as health and