Principal AI/ML HPC Specialist Technical Account Manager (STAM) , AWS Enterprise Support, NAMER-Sp

What you'll do

• Lead technical reviews and performance optimization for distributed AI/ML workloads and HPC clusters.
• Design and validate production training and inference architectures, including compute, networking, storage, orchestration, and monitoring.
• Advise enterprise customers and partner teams on AI/ML infrastructure, operational reliability, capacity planning, and best practices.

What they ask for

• Bachelor's degree
• 8+ years of experience in AI/ML, distributed computing, or GPU-accelerated infrastructure
• 3+ years of hands-on experience designing, implementing, or consulting on large-scale ML training or inference architectures in a customer-facing role
• 10+ years of IT development or implementation/consulting in the software, cloud computing, or AI/ML industries; experience with at least one major deep learning framework and ability to advise enterprise customers

Nice to have

• Distributed training and GPU/accelerator cluster experience, including frameworks such as PyTorch DDP, DeepSpeed, and Megatron-LM
• AWS Neuron SDK, SageMaker HyperPod, HPC schedulers, parallel file systems, and AWS certifications

Pay

182,800.00 - 247,300.00 USD annually (base salary)

Tools

AWS Parallel Computing Service (PCS), AWS ParallelCluster, PyTorch FSDP, DeepSpeed, SageMaker HyperPod, Amazon FSx for Lustre, AWS Neuron SDK, Elastic Fabric Adapter (EFA)

AI focus

Customer-facing design, deployment, and optimization of distributed AI/ML training and inference, including LLMs, HPC-to-AI workloads, GPU/Trainium clusters, and production MLOps.