AI Performance Engineer Intern: GPU & ML Systems

Hewlett Packard Enterprise; Spring (TX); Presencial; USD 48 000 - 56 000; Tempo parcial

Vantagens oferecidas por esta oferta de emprego

Health benefits

Career development programs

Inclusive culture

Resumo da oferta

Hewlett Packard Enterprise invites university students pursuing a technical Bachelor's degree to join our AI Performance Engineering Intern program. You will help design test plans, run experiments, and profile AI systems on GPU clusters using Docker/Apptainer.

Strong Python skills and knowledge of CUDA, ML frameworks, and HPC tools are required. This onsite internship offers exposure to cutting-edge AI workloads and collaboration with experienced engineers.

Qualificações

• Working toward B.S. in Computer Science, Applied Mathematics, Software Engineering or equivalent
• Coursework and experience with GPU hardware architectures and parallel programming techniques including NCCL/RCCL, NVSHMEM, or OpenSHMEM
• Background with AI applications and their programming models/languages, including PyTorch/TensorFlow
• Proficiency in Python
• Understanding of LLM parallelization methods TP/PP/DP, EP (MoE) is a plus
• Experience with AI clusters using job schedulers (e.g., PBS, Slurm) is a plus
• Experience with GPU clusters using Docker or Apptainer Containers is a plus
• Experience with GPU profiling using NVIDIA Nsight or AMD rocprof is a plus
• Proficiency with CUDA/ROCm is a plus
• Proficiency with Linux, scripting, and performance monitoring tools
• Experience with AI benchmarks (MLPerf) is a plus
• Strong organizational skills and attention to detail
• Excellent written and verbal communication skills
• Integrity, customer focus, teamwork and accountability in daily work

Responsabilidades

• Work with team members to design and create test plans to understand an AI system's performance
• Create and execute experiments to understand performance characteristics of the SUT
• Use profiling and monitoring tools to assist with performance characterization of the SUT
• Collaborate with team members on status, project progress, and issue resolution

Conhecimentos

Python

AI/ML concepts

CUDA/ROCm

Linux scripting

Performance profiling

TensorFlow/PyTorch

Team collaboration

Formação académica

Bachelor's degree in Computer Science or related field

Ferramentas

Docker

Apptainer

PBS/Slurm

NVIDIA Nsight

rocprof

Descrição da oferta de emprego

Hewlett Packard Enterprise invites university students pursuing a technical Bachelor's degree to join our AI Performance Engineering Intern program. You will help design test plans, run experiments, and profile AI systems on GPU clusters using Docker/Apptainer.

Strong Python skills and knowledge of CUDA, ML frameworks, and HPC tools are required. This onsite internship offers exposure to cutting-edge AI workloads and collaboration with experienced engineers.