GPU AI System Software Engineer Intern System Technologies and Engineering

TikTok · San Jose, California, USA

  • Internship

About the role

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

**Location** San Jose

**Employment Type** Intern

**Job Code** A211100

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 bottlenecks, debug and improve performance issues.

Minimum 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, Electrical Engineering, Computer Science or related majors.
  • Background with GPU/CPU benchmarking.
  • Familiar with ML/DL techniques, algorithms and frameworks like Tensor Flow 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.
  • Exposure to testing automation for various applications.
  • Experience with complex system‑level debugging is invaluable.

Job Information

The hourly rate range for this position in the selected city is $45- $45.

EEO Statement

For Los Angeles County (unincorporated) Candidates: Qualified applicants with arrest or conviction records will be considered for employment in accordance with all federal, state, and local laws including the Los Angeles County Fair Chance Ordinance for Employers and the California Fair Chance Act. Our company believes that criminal history may have a direct, adverse and negative relationship on the following job duties, potentially resulting in the withdrawal of the conditional offer of employment:

  • Interacting and occasionally having unsupervised contact with internal/external clients and/or colleagues;
  • Appropriately handling and managing confidential information including proprietary and trade secret information and access to information technology systems;
  • Exercising sound judgment.

#J-18808-Ljbffr

More at TikTok

More like this