Principal AI Specialist Solution Architect - In...
@ Amazon Web Services (AWS)

Beijing, Beijing, China
$240,000
On Site
Full Time
Posted 1 day ago

Your Application Journey

Personalized Resume
Apply
Email Hiring Manager
Interview

Email Hiring Manager

XXXXXXXXXX XXXXXXXXXXXXX XXXXXXXXX******* @amazon.com
Recommended after applying

Job Details

About AWS

Amazon Web Services (AWS) has been the most comprehensive and broadly adopted cloud platform since 2006. With over 240 fully featured services across compute, storage, databases, networking, analytics, machine learning, AI, IoT, and more, AWS supports virtually any workload for millions of customers globally.

Role Overview

The Principal AI Specialist Solution Architect - Infrastructure is a Subject Matter Expert in AI infrastructure. You will advise customers on model training and inference workloads, utilizing AWS accelerator computing services such as EC2, EKS, ECS, and SageMaker. Your deep technical expertise with GenAI, LLMs, Stable Diffusion, and state-of-the-art models will be key in driving customer success.

Key Responsibilities

  • Advise on optimal AI infrastructure solutions for model training and inference workloads.
  • Collaborate with Specialists and field teams to support GenAI practices and integration.
  • Develop white papers, blogs, presentations, and reference implementations.
  • Create enablement materials for AWS SA teams regarding GenAI solutions.
  • Engage directly with customers, partners, and senior engineers for technical guidance.

Basic & Preferred Qualifications

Minimum 5+ years of hands-on experience optimizing AI infrastructure. Advanced expertise in inference acceleration frameworks like vLLM, SGLang, and TensorRT, as well as distributed AI systems using frameworks such as Megatron-LM and Ray. Strong proficiency in Nvidia GPU performance optimization, CUDA programming and deep learning frameworks like PyTorch and TensorFlow. Preferred qualifications include experience with kernel programming, performance optimization on Amazon Trainiums and peer-reviewed technical publications.

About the Team

The team is built on mentorship and knowledge sharing with a diverse mix of experience levels. It emphasizes continuous career growth, robust support, and work/life balance.

Key skills/competency

  • AI Infrastructure
  • GenAI
  • LLM
  • Cloud Architecture
  • GPU Optimization
  • CUDA
  • Distributed Systems
  • SageMaker
  • White Papers
  • Customer Engagement

How to Get Hired at Amazon Web Services (AWS)

🎯 Tips for Getting Hired

  • Customize your resume: Highlight AI infrastructure and GenAI expertise.
  • Research AWS: Understand AWS cloud services and innovations.
  • Tailor your skills: Focus on GPU and distributed system experience.
  • Prepare technical examples: Share relevant project experiences.
  • Practice communication: Clear technical articulation is key.

📝 Interview Preparation Advice

Technical Preparation

Review CUDA programming best practices.
Study distributed training methodologies.
Practice GPU performance optimization techniques.
Refine knowledge in deep learning frameworks.

Behavioral Questions

Describe a challenging AI project.
Explain a time of rapid problem-solving.
Discuss collaborative customer engagements.
Share an example of effective mentorship.

Frequently Asked Questions