Model Infra Software Engineer Intern: ML/LM Systems

TikTok; San Jose (CA); On-site; USD 51,143 - 72,840; Full time

Benefits offered by this job

Health insurance

Life insurance

Well-being benefits

10 paid holidays per year

Paid sick time

Job summary

A leading social media platform in California is seeking a Software Engineering Intern for the Model Infra team. The role involves optimizing recommendation systems, working on generative AI model integration, and driving hardware efficiency. Candidates should be pursuing a degree in Software Development or a related field, have strong programming skills in C++ and Python, and possess knowledge of computer architecture. Interns receive competitive hourly compensation starting at $45, benefits including health insurance, and paid holidays.

Qualifications

• Currently pursuing an Undergraduate/Master in a relevant technical discipline.
• Strong programming skills in C++ and Python.
• Solid understanding of Computer Architecture and the GPU software stack.
• Experience with deep learning frameworks and eager to investigate model execution.

Responsibilities

• Drive optimization of training and inference pipelines.
• Architect specialized systems for LLM integration.
• Build high-concurrency engines for Petabyte-scale streaming.
• Collaborate with researchers on next-generation recommendation architectures.
• Innovate on storage and synchronization of massive model states.

Skills

Strong programming skills in C++

Strong programming skills in Python

Solid understanding of Computer Architecture

Understanding of GPU software stack (CUDA)

Experience with deep learning frameworks (PyTorch, TensorFlow)

Education

Undergraduate/Master in Software Development, Computer Science, Computer Engineering

Job description

A leading social media platform in California is seeking a Software Engineering Intern for the Model Infra team. The role involves optimizing recommendation systems, working on generative AI model integration, and driving hardware efficiency. Candidates should be pursuing a degree in Software Development or a related field, have strong programming skills in C++ and Python, and possess knowledge of computer architecture. Interns receive competitive hourly compensation starting at $45, benefits including health insurance, and paid holidays.