AI Performance Engineering Intern

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 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 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 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 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 work during school breaks and then return 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.

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 @HPECareers on Instagram to see the latest on people, culture and tech at HPE.