AI/ML HPC Principal TAM for Enterprise

Amazon Web Services (AWS); Chicago (IL); On-site; USD 183,000 - 247,000; Full time

Job summary

Amazon Web Services, Inc. in NAMER region seeks a Principal AI/ML HPC Specialist to join the TAM team. You will lead strategic technical engagements, guide NAMER labs and enterprise customers through AI/ML transformation journeys, and architect scalable GPU/Trainium cluster solutions.

You'll work across Slurm, SageMaker HyperPod, DLAMIs, NCCL tuning, and cost governance, helping customers optimize training and inference at scale while partnering with TAMs and service teams.

Qualifications

• 8+ years in Data & AI technologies including AI/ML, analytics, storage or databases.
• 3+ years designing or consulting on large-scale ML training or inference architectures.
• 10+ years IT development or consulting in software/cloud/AI/ML industries.
• Ability to serve as a trusted technical advisor to enterprise customers.

Responsibilities

• Lead deep technical engagements and optimize enterprise AI/ML workloads across GPU clusters.
• Design production-grade AI/ML training and inference using Slurm, PyTorch FSDP, DeepSpeed, Megatron-LM, and SageMaker HyperPod.
• Support customers in building ML solutions including LLMs, PINNs, and MLOps pipelines.
• Improve training throughput, NCCL communications, and GPU utilization while ensuring governance and cost controls.
• Partner with service teams to guide AI/ML infrastructure decisions spanning compute, networking, storage, and orchestration.

Skills

AI/ML Expertise

Cloud computing

Enterprise engagement

Tools

PyTorch

TensorFlow

MxNet

SageMaker HyperPod

ParallelCluster

DLAMI

NCCL

AWS Neuron SDK

Slurm

Job description

Amazon Web Services, Inc. in NAMER region seeks a Principal AI/ML HPC Specialist to join the TAM team. You will lead strategic technical engagements, guide NAMER labs and enterprise customers through AI/ML transformation journeys, and architect scalable GPU/Trainium cluster solutions.

You'll work across Slurm, SageMaker HyperPod, DLAMIs, NCCL tuning, and cost governance, helping customers optimize training and inference at scale while partnering with TAMs and service teams.