Jobs · Engineering · California

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

TikTok · San Jose, CA · 1 wk ago
Engineering$45/hrInternship

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.

Qualifications

  • 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 Qualifications

  • 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.
  • Experience with complex system-level debugging is invaluable.

Job Information

[For Pay Transparency] Compensation Description (Hourly) - Campus Intern
The hourly rate range for this position in the selected city is $45- $45.
Benefits may vary depending on the nature of employment and the country work location. Interns have day one access to health insurance, life insurance, wellbeing benefits and more. Interns also receive 10 paid holidays per year and paid sick time (56 hours if hired in first half of year, 40 if hired in second half of year). Interns who are not working 100% remote may also be eligible for housing allowance.

About TikTok

TikTok is the leading destination for short-form mobile video. At TikTok, our mission is to inspire creativity and bring joy. TikTok's global headquarters are in Los Angeles and Singapore, and we also have offices in New York City, London, Dublin, Paris, Berlin, Dubai, Jakarta, Seoul, and Tokyo.

Diversity & Inclusion

TikTok is committed to creating an inclusive space where employees are valued for their skills, experiences, and unique perspectives. Our platform connects people from across the globe and so does our workplace. At TikTok, our mission is to inspire creativity and bring joy. To achieve that goal, we are committed to celebrating our diverse voices and to creating an environment that reflects the many communities we reach. We are passionate about this and hope you are too.

Similar jobs