Apply on Employer Site

TikTok · 2 weeks ago

GPU/AI Application System Software Engineer Intern (System Technologies and Engineering) - 2026 Summer (BS/MS)

San Jose, CA

Internship

Onsite

Intern

$45/hr - $45/hr

Start in 2026 Spring

TikTok is the leading destination for short-form mobile video, and they are seeking a GPU/AI Application System Software Engineer Intern for Summer 2026. The role involves developing optimized software for AI and HPC workloads, performance benchmarking, and collaboration with internal teams to enhance system performance.

Content CreatorsContent DiscoveryMedia and EntertainmentSocial MediaVideo

H1B Sponsor Likely

Responsibilities

Design and implement performance benchmarks and testing methodologies to evaluate system performance

Develop benchmark tools and performance optimization of AI workloads specifically tailored for large-scale LLM training and inference, as well as High-Performance Computing (HPC)

Develop Python scripts to automate the testing of various benchmark tools

Collaborate with internal teams to identify system bottleneck, debug and improve performance issues

Qualification

GPU/CPU benchmarkingML/DL frameworksPythonC/C++Linux systemsHigh Performance ComputingAI model developmentParallel programmingCUDA programmingTesting automationGit workflowComplex system-level debugging

Required

Must be able to commit to a 12-week full-time work period during Summer 2026

Currently pursuing a Bachelor's, Master's, or PhD degree within Computer Engineering in Electrical Engineering, Computer Engineering, Computer Science or related majors

Background with GPU/CPU benchmarking

Familiar with ML/DL techniques, algorithms and frameworks like TensorFlow or PyTorch

Exposure to testing automation for various applications

Proficiency in Python and C/C++

Hands-on experience with Linux based systems

Ability to work independently and complete projects from beginning to end and in a timely manner

Preferred

Strong background in one of the following fields: High Performance Computing, ML Hardware Acceleration (e.g., GPU/TPU/RDMA) or ML for Systems, and Distributed Storage

Experience in AI model development, training, evaluation and deployment on Cloud, Cluster or on-premises

Experience with parallel programming and at least one communication runtime (MPI, NCCL, UCX, NVSHMEM)

Experience with development applications with CUDA programming

Linux kernel development experience, such as networking and device drivers etc

Familiar with git workflow

Exposure to testing automation for various applications

Experience with complex system-level debugging is invaluable

Benefits

Interns have day one access to health insurance

Life insurance

Wellbeing benefits

10 paid holidays per year

Paid sick time (56 hours if hired in first half of year, 40 if hired in second half of year)

Housing allowance

Company

TikTok

Glassdoor3.4

TikTok is a short-form video entertainment app and social network platform. It is a sub-organization of ByteDance.

Founded in 2003

Los Angeles, California, USA

10001+ employees

https://www.tiktok.com

H1B Sponsorship

TikTok has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (979)

2024 (601)

2023 (387)

2022 (322)

2021 (133)

2020 (72)