Principal AI/ML HPC TAM - Flexible Hours

Amazon Inc.; San Francisco (CA); On-site; USD 210,000 - 284,000; Full time

Job summary

Amazon Web Services, Inc. is seeking a Principal Technical Account Manager for AWS Enterprise Support NAMER-Sp to guide enterprise customers through AI/ML transformation.

You will lead strategic technical engagements, architect solutions, and ensure performance and cost governance at scale. You’ll work with TAMs, SAs, and service teams to deliver best-practice guidance on HPC-to-AI convergence, distributed training, and production-grade inference, enabling customers to innovate with AWS AI

Qualifications

• 8+ years of working with Data & AI related technologies, including, but not limited to, AI/ML, Analytics, Database, and/or Storage experience
• 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
• Demonstrated ability to serve as a trusted technical advisor to enterprise customers

Responsibilities

• Deliver Strategic Technical Engagements - Lead comprehensive technical deep-dives and performance optimization for enterprise AI/ML workloads including distributed training cluster architecture using AWS Parallel Computing Service (PCS) and AWS ParallelCluster, the latest GPU-accelerated computing (i.e., P6/P6e, G7/G7e instances), AWS Trainium-based training (Trn3 UltraServers), and multi-node NCCL communication tuning over EFA’s SRD protocol.
• Architect and Validate Innovative Solutions - Design and implement production-grade AI/ML training and inference solutions leveraging Slurm-based job scheduling, distributed training frameworks (PyTorch FSDP, DDP, DeepSpeed, Megatron-LM), SageMaker HyperPod for managed GPU clusters with automated health checks and node replacement, high-performance parallel storage (Amazon FSx for Lustre), and container runtimes on Deep Learning AMIs (DLAMIs) against reference architectures and HPC lens to ensure performance, reliability, and cost governance at scale.
• Enable Customer Success - Support customers in implementing business-critical HPC capabilities including the development of large language model (LLM) (Llama, GPT-class models), physics-informed neural networks (PINNs) and surrogate models, MLOps pipelines, simulation-ML hybrid architectures orchestrated by AWS Step Functions and AWS Batch, distributed data processing, cluster observability, and governance controls for GPU/Trainium-intensive workloads.
• Enable Business Critical Outcomes - Partner with service teams to enhance model training throughput, optimize NCCL collective communications, improve GPU/Trainium utilization across multi-node UltraClusters, and drive operational efficiency through proactive monitoring, automated failure recovery (HyperPod health checks), and capacity planning (EC2 Capacity Blocks for ML). Contribute to product roadmap, share reference architecture performance and benchmarks with broader TAM and Technical communities.
• Serve as Trusted Advisor and Advocate — Develop and nurture technical partnerships with enterprise stakeholders, serving as the trusted advisor for AI/ML infrastructure decisions spanning compute, networking (Elastic Fabric Adapter with SRD), storage, orchestration, and the HPC-to-AI convergence journey.

Job description

Amazon Web Services, Inc. is seeking a Principal Technical Account Manager for AWS Enterprise Support NAMER-Sp to guide enterprise customers through AI/ML transformation.

You will lead strategic technical engagements, architect solutions, and ensure performance and cost governance at scale. You’ll work with TAMs, SAs, and service teams to deliver best-practice guidance on HPC-to-AI convergence, distributed training, and production-grade inference, enabling customers to innovate with AWS AI