Technical Account Manager - Ai/ml 北京、广州、深圳 jobs at Amazon in China, Beijing

Key job responsibilities
- As an AIML Specialist Solutions Architect (SA) in AI Infrastructure, you will serve as the Subject Matter Expert (SME) for providing optimal solutions in model training and inference workloads that leverage Amazon Web Services accelerator computing services. As part of the Specialist Solutions Architecture team, you will work closely with other Specialist SAs to enable large-scale customer model workloads and drive the adoption of AWS EC2, EKS, ECS, SageMaker and other computing platform for GenAI practice.
- You will interact with other SAs in the field, providing guidance on their customer engagements, and you will develop white papers, blogs, reference implementations, and presentations to enable customers and partners to fully leverage AI Infrastructure on Amazon Web Services. You will also create field enablement materials for the broader SA population, to help them understand how to integrate Amazon Web Services GenAI solutions into customer architectures.
- You must have deep technical experience working with technologies related to Large Language Model (LLM), Stable Diffusion and many other SOTA model architectures, from model designing, fine-tuning, distributed training to inference acceleration. A strong developing machine learning background is preferred, in addition to experience building application and architecture design. You will be familiar with the ecosystem of Nvidia and related technical options, and will leverage this knowledge to help Amazon Web Services customers in their selection process.
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.Mentorship & Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Work/Life Balance

- 5+ years of hands-on experience optimizing AI infrastructure, with deep expertise in inference acceleration frameworks (e.g., vLLM, SGLang, TensorRT, etc.), model training and serving systems across PyTorch and TensorFlow ecosystems;
- Advanced proficiency in Nvidia GPU performance optimization techniques, including memory management, kernel fusion, and quantization strategies for large-scale deep learning workloads;
- Strong foundation in parallel computing principles with practical CUDA programming experience, emphasizing efficient resource utilization and throughput maximization;
- Demonstrated success implementing and tuning distributed AI systems leveraging modern frameworks like Megatron-LM and Ray, with particular focus on LLM deployment and horizontal scaling across GPU clusters.

Key job responsibilities
- Act as a thought leader sharing best practices through forums such as AWS blogs, whitepapers, reference architectures and public-speaking events such as AWS Summit, AWS re: Invent, etc.We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.AWS values diverse experiences. Even if you do not meet all of the qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.

- 5+ years design and implementation experience with NoSQL technologies and Key-value stores such as ElastiCache, Aerospike, Redis, Memcached, including NoSQL data modeling.
- 3+ years software architecture and development experience using languages such as Java, Python, Go, and Node.js
- - A degree in computer science, software engineering, or mathematics or 10 years equivalent work experience.
- - Knowledge of the underlying infrastructure requirements such as Networking, Storage, and Hardware Optimization.
- - Demonstrated effectiveness working across multiple business units to achieve results.
- - An understanding of computing deployment models as they relate to databases and analytics.
- - Working knowledge of modern software development practices such as agile methodologies and DevOps.

Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

• Responsible for the over-all systems development life cycle.
• Management and execution against project plans and delivery commitments; Manage the day-to-day activities of the engineering team within an Agile/Scrum environment.
• Management of departmental resources, staffing, mentoring, and enhancing and maintaining a best-of-class engineering team.
• Work closely with the engineers to architect and develop the best technical design and approach.
• Report on status of development, quality, operations, and system performance to management.
Key job responsibilities
• Responsible for the over-all systems development life cycle.
• Management and execution against project plans and delivery commitments; Manage the day-to-day activities of the engineering team within an Agile/Scrum environment.
• Management of departmental resources, staffing, mentoring, and enhancing and maintaining a best-of-class engineering team.
• Work closely with the engineers to architect and develop the best technical design and approach.
• Report on status of development, quality, operations, and system performance to management.

- Bachelor's Degree in Computer Science or related field.
- 10+ years’ experience managing software development teams, ideally with some hands-on architectural or distributed systems experience.
- The key requirement for this position is established skill designing and developing complex, interactive end user experiences.

- MS, Ph.D. in Computer Science, Computer Engineering, Electrical Engineering, etc.
- A strong track record of project delivery for large and cross-functional projects.
- Strong technical credentials and deep hands-on technical expertise on embedded systems.
- Development experience on mobile platforms such as Google Android, WebOS, Windows Mobile, Apple iPhone and RIM Blackberry.
- Excellence in technical communication with peers and non-technical cohorts.
- Ability to handle multiple competing priorities in a fast-paced environment.
- Excellent written and verbal communication skills with the ability to present complex technical information in a clear and concise manner to a variety of audiences.
- A track record of communicating well with executives and non-technical leaders.
- Experienced on smart TV products.

*Hiring location: Beijing, Shanghai, Guangzhou, Shenzhen, Hong Kong(visa sponsorship provided)As a Senior Technical Account Manager (Sr. TAM) specializing in GPU Optimization in AWS Enterprise Support, you will play a crucial role in two key missions: guiding customers' GPU acceleration initiatives across AWS's comprehensive compute portfolio, and spearheading the development of optimization strategies that revolutionize customer workload performance.Key Job Responsibilities
- Design and optimize GPU resource usage on EC2/EKS/SageMaker or equivalent cloud compute, container, and ML services; implement node pool tiering, Karpenter/Cluster Autoscaler tuning, auto scaling, and cost governance (Savings Plans/RI/Spot/ODCR or equivalent).
- Drive GPU partitioning and multi-tenant resource sharing strategies to reduce idle resources and increase overall cluster utilization.- Build GPU observability and monitoring systems (nvidia-smi, CloudWatch or equivalent monitoring tools, profilers, distributed communication metrics) to align capacity planning with SLOs.
- Ensure compatibility across GPU drivers, CUDA, container runtimes, and frameworks; standardize change management and rollback processes.
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.
Mentorship & Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Work/Life Balance

- 5+ years in cloud technical support, solutions architecture, or customer success management, with at least 3 years of hands-on experience in GPU/accelerated computing platforms.
- In-depth understanding of GPU instance families (e.g., AWS G/P/H series) or similar offerings from other cloud providers, AMI/driver/CUDA/container compatibility management, and cloud storage/network performance tuning (e.g., S3 I/O, EBS/Instance Store equivalents, preprocessing pipelines). Proficient in scheduling GPU workloads with EKS or equivalent Kubernetes-based orchestration services, including node pool tiering, resource quotas, elastic scaling, and auto-recovery strategies. Experienced in multi-GPU/multi-node distributed computing (NCCL, topology awareness, tensor parallelism, pipeline parallelism) with expertise in communication optimization for large-scale AI training and inference.
- Skilled in PyTorch/TensorFlow performance analysis and optimization, including DataLoader tuning, mixed precision, operator fusion, and inference acceleration toolchains (ONNX, TensorRT, CUDA Graphs).
- Experienced in cost and capacity governance, familiar with Savings Plans, RI, ODCR, Spot, Capacity Blocks, and right-sizing strategies or their equivalents in other cloud platforms.
- Demonstrated cross-functional communication and influence skills, capable of driving technical solutions with data and business objectives.

- AWS Solutions Architect Professional, Machine Learning Specialty, or DevOps Professional certification or equivalent credentials from other cloud providers.
- Hands-on experience with NVIDIA ecosystem software and toolchains (CUDA/cuDNN/NCCL, TensorRT, CUDA Graphs) and proven ability to maintain performance consistency across versions and platforms.
- Delivered quantifiable performance improvements (GPU throughput, latency reduction, cost savings) with demonstrated benchmarking and regression testing methodology.
- Proven repeatable optimization results in LLM inference, batch AI training, real-time video processing, or high-performance computing (HPC).
- Contributions to open source projects (Run:ai, Ray, vLLM, DeepSpeed, Kubeflow, etc.) or published technical articles, whitepapers, or performance benchmarking.
- Experience with Infrastructure as Code (Terraform, AWS CDK **or equivalent cloud development frameworks**), Helm Charts, baseline container image management, and DevOps automation.
- Able to present performance-business tradeoffs and results to senior stakeholders using PR/FAQ documents, architecture diagrams, and capacity/cost reports.

Every day will bring new and exciting challenges on the job while you:- Make recommendations on AI architecture, security, cost optimization, and operational excellence- Complete analysis and present periodic reviews of AI workload performance- Foster an ecosystem where AI and humans progress together through knowledge sharingDiverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.
Mentorship & Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Work/Life Balance

- Master's degree in Computer Science, Artificial Intelligence, Machine Learning or related technical field
- Deep understanding of Generative AI fundamentals, Large Language Models, and core AI/ML concepts
- Excellent theoretical foundation in AI technologies and their underlying principles
- Excellent empirical reasoning skills with ability to formulate theories, make predictions, validate and refine solutions

- PhD in Computer Science, Artificial Intelligence, Machine Learning or related field
- Demonstrated ability to build systematic knowledge frameworks across technology stacks
- Experience with implementing enterprise-scale AI/ML solutions
- Experience with AWS core services across Compute, Storage, Database, and Networking
- Knowledge of cloud architecture patterns and distributed systems, cloud security principles and compliance frameworks
- Experience with container technologies and serverless architectures, modern software development practices and DevOps
- Excellence in stakeholder management and technical communication

Key job responsibilities
• Maintain close connection with WW Marketplace COE Teams to ensure that China requirements are regularly communicated and understood.
• Maintain close connection with Area stakeholders to have a deep understand what customers and partners are requiring in order to make AWSMP fit for purpose.
• Lead the team of specialists and motivate them to execute exceptionally well as part of the China partner organization.
A day in the life
This leader will work towards ensuring they keep in close touch with what each Area in China requires to reach mass adoption of AWS Marketplace by the prevalent ISVs and Channel Partners in the market. They use this understanding to direct the team to build and execute programs to suit each area's unique requirements.About the team
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.
Mentorship & Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Work/Life Balance

• 10+ years in business development, market development, strategic partnerships, or program/product management relating to infrastructure and cloud technology.
• Proven people and team management experience; experience working with new product and service development teams, and outbound functions.
• Demonstrated ability to engage and influence senior stakeholders and executive teams.
• Track record of building rapport with senior customer executives (e.g. CEO, CIO, CTO), building close relationships, and closing opportunities with key, strategic accounts.
• Experience and success in negotiating complex deals with customers and partners; strong verbal and written communications skills in English.

• MBA/Technical Master’s or Graduate Degree
• Prior experience growing software business
• Experience with technology – cloud computing, SaaS, PaaS, etc – would be a plus

Key job responsibilities
- Delivery of high-quality, scalable software on-time and on-budget.- Be deeply involved with engineers to architect and develop the best technical designs and approaches.- Build operational excellence to reduce lead times and maintain the operations of critical services.- Clearly communicate technical decisions and progress to both technical and non-technical audiences.- Translate business, functional, and technical requirements into detailed project plans.A day in the life
About the team
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.Mentorship & Career Growth
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.Work/Life Balance

- 3+ years of engineering team management experience and 7+ years of engineering experience
- 8+ years of leading the definition and development of multi tier web services experience
- Knowledge of engineering practices and patterns for the full software/hardware/networks development life cycle, including coding standards, code reviews, source control management, build processes, testing, certification, and livesite operations
- Experience partnering with product or program management teams and experience delivering products against plan in a fast-paced, multi-disciplined, distributed-responsibility and often ambiguous environment
- Experience designing or architecting (design patterns, reliability and scaling) of new and existing systems