AI Performance Engineering Intern

Hewlett Packard Enterprise Development LP; Town of Texas, Northern (WI, KY); Hybrid; USD 48,000 - 56,000; Full time

Benefits offered by this job

Health & Wellbeing

Personal & Professional Development

Unconditional Inclusion

Job summary

Hewlett Packard Enterprise Development LP is seeking an AI Performance Engineering Intern to design test plans and run experiments that quantify AI system performance. You will use profiling tools and work with GPUs, including CUDA and mixed-precision techniques, in Linux environments.

The role requires coursework in CS/Math/Software Eng and strong Python skills, plus familiarity with PyTorch/TensorFlow and AI hardware concepts.

Qualifications

• Working toward a B.S. in Computer Science, Applied Mathematics, Software Engineering or equivalent.
• Coursework and experience with GPU (NVIDIA/AMD) hardware architectures and parallel programming techniques, including parallel programming methods such as NCCL/RCCL, NVSHMEM, or OpenSHMEM
• Background with AI applications and their programming models/languages, including PyTorch/TensorFlow
• Proficiency in Python
• Deep understanding of LLM parallelization methods TP/PP/DP, and especially EP (MoE) is a plus
• Experience with using inference serving engines (e.g. vLLM, Ollama) is a plus
• Experience with AI clusters using job schedulers (e.g., PBS, Slurm) is a plus
• Experience on GPU-based AI clusters with 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 the Linux operating systems, scripting, and knowledge of performance monitoring tools
• Experience with AI industry standard benchmarks (MLPerf) is a plus
• Self-driven, solid work ethic, and motivation
• Ability to multi-task with strong organizational skills and attention to detail
• Excellent analytical and problem-solving skills
• Integrity, customer focus, innovation, teamwork, and accountability in daily work
• Ability to work in a fast-paced, multitasking environment
• Excellent written and verbal communication skills. Mastery of the English language is required.

Responsibilities

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

Skills

Python

CUDA

PyTorch/TensorFlow

GPU architectures

Linux

NCCL/RCCL

NVSHMEM

OpenSHMEM

Education

Working toward B.S. in Computer Science, Applied Mathematics, Software Engineering or equivalent

Tools

Docker

Apptainer

NVIDIA Nsight

AMD rocprof

PBS

Slurm

Job description

AI Performance Engineering InternThis role has been designed as ''Onsite' with an expectation that you will primarily work from an HPE office.**Who We Are:**Hewlett Packard Enterprise is the global edge-to-cloud company advancing the way people live and work. We help companies connect, protect, analyze, and act on their data and applications wherever they live, from edge to cloud, so they can turn insights into outcomes at the speed required to thrive in today's complex world. Our culture thrives on finding new and better ways to accelerate what's next. We know varied backgrounds are valued and succeed here. We have the flexibility to manage our work and personal needs. We make bold moves, together, and are a force for good. If you are looking to stretch and grow your career our culture will embrace you. Open up opportunities with HPE.**Job Description:*****Responsibilities:**** Work with team members to design and create test plans to understand an AI system's performance.* Create and execute experiments to understand the performance characteristics of the system under test (SUT).* Use profiling and/or monitoring tools to assist with the performance characterization of the SUT.* Collaborate with team members regarding status, project progress, and issue resolution.***Education and Experience Required:**** Working toward B.S. in Computer Science, Applied Mathematics, Software Engineering or equivalent* Coursework and experience with GPU (NVIDIA/AMD) hardware architectures and parallel programming techniques, including parallel programming methods such as NCCL/RCCL, NVSHMEM, or OpenSHMEM**Knowledge and Skills:*** Background with AI applications and their programming models/languages, including PyTorch/TensorFlow* Proficiency in Python* Deep understanding of LLM parallelization methods TP/PP/DP, and especially EP (MoE) is a plus* Experience with using inference serving engines (e.g. vLLM, Ollama) is a plus* Experience with AI clusters using job schedulers (e.g., PBS, Slurm) is a plus* Experience on GPU-based AI clusters with 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 the Linux operating systems, scripting, and knowledge of performance monitoring tools* Experience with AI industry standard benchmarks (MLPerf) is a plus* Self-driven, solid work ethic, and motivation* Ability to multi-task with strong organizational skills and attention to detail* Excellent analytical and problem-solving skills* Integrity, customer focus, innovation, teamwork, and accountability in daily work* Ability to work in a fast-paced, multitasking environment* Excellent written and verbal communication skills. Mastery of the English language is required.***Impact/Scope:**** University students actively enrolled in a university/college/institution pursuing a Bachelor's degree in a technical area of study, and who typically works during school breaks and then returns to their university. While this applies to most interns, there may be opportunities for full-time students to work as part-time interns during the school year in some countries.**Accessibility**HPE is committed to creating an inclusive and accessible workplace and encourages applications from all qualified individuals, including those with disabilities. If you believe you require accommodation during any stage of the application or interview process, please submit your request by completing our secure form linked .Note: This option is reserved for applicants needing assistance/reasonable accommodation related to a disability.**What We Can Offer You:****Health & Wellbeing**We strive to provide our team members and their loved ones with a comprehensive suite of benefits that supports their physical, financial and emotional wellbeing.**Personal & Professional Development**We also invest in your career because the better you are, the better we all are. We have specific programs catered to helping you reach any career goals you have - whether you want to become a knowledge expert in your field or apply your skills to another division.**Unconditional Inclusion**We are unconditionally inclusive in the way we work and celebrate individual uniqueness. We know varied backgrounds are valued and succeed here. We have the flexibility to manage our work and personal needs. We make bold moves, together, and are a force for good.**Let's Stay Connected:**Follow on Instagram to see the latest on people, culture and tech at HPE.#unitedstates#######interns**Job:**Administration**Job Level:**N/A\"The expected salary/wage range for this position is provided below. Actual offer may vary from this range based upon geographic location, work experience, education/training, and/or skill level.- United States of America: Hourly Salary USD 35.00 - 41.00 in TexasThe listed salary range reflects base salary. Variable incentives may also be offered.\"Information about employee benefits offered in the US can be found atHPE is an Equal Employment Opportunity/ Veterans/Disabled/LGBT employer. We do not discriminate on the basis of race, gender, or any other protected category, and all decisions we make are made on the basis of qualifications, merit, and business need. Our goal is to be one global team that is representative of our customers, in an inclusive environment where we can continue to innovate and grow together. Please click here: .**Hewlett Packard Enterprise is EEO Protected Veteran/ Individual with Disabilities.**HPE will comply with all applicable laws related to employer use of arrest and conviction records, including laws requiring employers to consider for employment qualified applicants with criminal histories.**Recruitment Fraud Alert**We have become aware of an increase in fraudulent recruitment activities in which individuals impersonate our company or authorized recruitment agencies to offer fake employment opportunities. These scams may occur through false websites, emails, social media, or chat-based applications and often aim to obtain personal information or money. Please note that Hewlett Packard Enterprise (HPE), its direct and indirect subsidiaries and affiliated companies, and its authorized recruitment agencies/vendors will never charge a candidate a registration fee, hiring fee, or any other fee in connection with its recruitment and hiring process. We also never request personal information such as back account details, Social Security numbers, or national IDs via social media or chat applications.All legitimate job opportunities will come through official company channels, and candidates are responsible for verifying the credentials of any third party claiming to represent the company. Any reliance on fraudulent communication is at the individual's own risk, and HPE disclaims legal liability for any resulting damages. If you suspect recruitment fraud, do not share personal information or make any payments and report the incident to your local authorities immediately.Please ensure the resume you submit to us does not include any sensitive personal data. Sensitive personal data includes data revealing information about your racial or ethnic origin, political opinions, religious or philosophical beliefs, trade union membership, health, sex life or sexual orientation. To the extent the resume you submit does contain this type of personal data, you consent to the storing and processing of this data by HPE for the purpose of reviewing and managing your application.