AI and Machine Learning Engineer

Key Skills

Aartificial intelligence, Analytical and Problem solving, CUDA kernels, Deep Learning, Design, DevOps, Effective communication skills, GPUs, High performance computing- HPC, IT, Linux Operating system, Machine learning techniques, Python Programming, PyTorch, TensorFlow

Job Description

We are seeking a AI/Machine Learning (ML) Performance Engineer with experience in optimizing and tuning ML workloads on a variety of platforms. Familiarity with Linux operating systems, containers and extensive experience working in cluster(multi-server) IT environments in the cloud and on premise. Experience working with communication libraries such as MPI, UCX and the ability to build ML workloads from the ground-up to run within HPC Linux clusters. This includes pre-containerizing the workload, building the container and then deploying the container on the cluster. Requires deep knowledge of tuning and optimizing various system parameters to get optimal performance on the system and cluster.

Responsibilities

Applicants should be able to write a neural network in python and utilize profiling tools to figure out the bottlenecks in the performance. Knowledge of accelerators like Graphic Processing Units (GPUs) and their associated libraries such as NCCL, CUDA and cudNN.

Most importantly, we seek applicants who are excited to work at the cutting edge of technology with the most capable and talented lab team in AI. Strong written and spoken communication skills are a must so we can maintain high efficiency on a remote, globally dispersed team.

Responsibilities

• Optimize the performance of deep learning models on GPUs and accelarators.
• Analyze and optimize the performance of deep learning models at scale.
• Keep yourself up to date with the latest in the AI/machine learning field and share this knowledge with the team.
• Port/Optimize the latest models coming out of research on to the HPE platform.

Required Skills and Experience 

• Must have experience with Linux command-line on multiple Linux distributions
• Experience coding in Python and reading/understanding python code.
• Experience with machine and deep learning frameworks such as PytorchPyTorch, TensorflowTensorFlow and MXNet.
• Experience with InfinibandInfiniBand technology
• Must have strong problem analysis and problem solving skills
• Experience with one or more scripting languages.

Additional Skills:

Cloud Architectures, Cross Domain Knowledge, Design Thinking, Development Fundamentals, DevOps, Distributed Computing, Microservices Fluency, Full Stack Development, Security-First Mindset, Solutions Design, Testing & Automation, User Experience (UX)

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.

Diversity, Inclusion & Belonging

We are unconditionally inclusive in the way we work and celebrate individual uniqueness. We know diverse 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.

#unitedstates

#highperformancecompute

Job:

Engineering

Job Level:

Expert

States with Pay Range Requirement

The expected salary/wage range for a U.S.-based hire filling this position is provided below. Actual offer may vary from this range based upon geographic location, work experience, education/training, and/or skill level. If this is a sales role, then the listed salary range reflects combined base salary and target-level sales compensation pay. If this is a non-sales role, then the listed salary range reflects base salary only. Variable incentives may also be offered. Information about employee benefits offered can be found at https://myhperewards.com/main/new-hire-enrollment.html.

Annual Salary: $124,500.00 - $286,300.00