refresh

热门公司

Trending

招聘

JobsAMD

Senior AI/ML and GPU Performance Analysis Engineer

AMD

Senior AI/ML and GPU Performance Analysis Engineer

AMD

Hyderabad

·

On-site

·

Full-time

·

1mo ago

Benefits & Perks

Wellness benefits

Top Tier compensation with equity

Health, dental, and vision coverage

Flexible PTO policy

Remote work flexibility

Required Skills

SQL

TensorFlow

Python

WHAT YOU DO AT AMD CHANGES EVERYTHING:

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond.  Together, we advance your career.AI/ML and GPU Performance QA engineer
We are seeking an experienced Senior Technical Validation Engineer to drive validation and performance engineering for Machine Learning (ML), High-Performance Computing (HPC) frameworks, GPU software stacks, and cluster environments.

This role requires good understanding and experience in ROCm, CUDA, GPU architecture, ML frameworks, CI/CD systems, benchmarking, and competitive analysis.
You will lead cross-functional initiatives across validation, automation, test development, performance tuning, and system scalability, ensuring delivery of high-quality, high-performance software for next-generation AI and HPC workloads.

Key Responsibilities:

  • Lead validation for ML/AI models: accuracy testing, performance benchmarking, regression, drift detection, A/B testing
  • Test ML frameworks: Py Torch, Hugging Face, MLFlow experiment tracking
  • Validate wide varieties of AI models to ensure correctness in distributed training or inference
  • Perform GPU testing & profiling: ROCm/CUDA validation, performance profiling, memory/thermal analysis, multi-GPU scaling
  • Validate HPC frameworks, distributed runtimes, compilers, and GPU libraries
  • Build scalable CI/CD workflows for ML/HPC validation. Develop automated test pipelines using Docker, Kubernetes, GitHub Actions, Jenkins
  • Validate cloud-based AI workloads on AWS Sage Maker, Lambda, and S3
  • Test the benchmarks under containerized and virtualized GPU environments
  • Design and implement automated validation pipelines for ML frameworks (e.g., Py Torch, Tensor Flow, JAX) across GPU platforms.
  • Develop and maintain benchmarking suites for AI models and HPC workloads, focusing on performance, scalability, and regression detection.
  • Multi-node validation efforts using orchestration tools (e.g., Slurm, MPI, Kubernetes) to simulate real-world distributed training and inference.
  • Collaborate with hardware and software teams to validate GPU hardware platforms (NVIDIA CUDA, AMD ROCm) for ML and HPC readiness.
  • Analyze performance metrics using profiling tools (e.g., Nsight, rocprof, perf) and provide actionable insights.
  • Drive test content development for emerging AI workloads, including LLMs, vision models, and scientific computing benchmarks.
  • Perform bottleneck analysis, hyperparameter validation, and competitive benchmarking
  • Mentor junior engineers and contribute to validation strategy, tooling, and best practices.

Preferred Expereince :

  • Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or related field.
  • 8+ years of experience in validation engineering, ML infrastructure, or HPC performance testing.
  • Strong hands-on experience with GPU platforms (NVIDIA CUDA, AMD ROCm) and their software ecosystems.
  • Deep understanding of AI model architectures, training/inference workflows, and ML performance bottlenecks.
  • Proven experience with CI/CD systems, Git, Docker, and automated test frameworks.
  • Expertise in multi-node orchestration and distributed system validation.
  • Familiarity with HPC benchmarks (e.g., HPL, HPCG, MLPerf) and AI model benchmarking methodologies.
  • Proficiency in scripting and automation (Python, Bash, YAML) in Linux environments.
  • Strong communication, documentation, and cross-functional collaboration skills.

Benefits offered are described:  AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law.   We encourage applications from all qualified candidates and will accommodate applicants’ needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position.  AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.

Total Views

0

Apply Clicks

0

Mock Applicants

0

Scraps

0

About AMD

AMD

AMD

Public

A semiconductor company that designs and develops graphics units, processors, and media solutions

10,001+

Employees

Santa Clara

Headquarters

Reviews

3.5

25 reviews

Work Life Balance

3.2

Compensation

4.1

Culture

3.6

Career

3.4

Management

3.1

65%

Recommend to a Friend

Pros

Good compensation and benefits

Positive work environment

Great management and coworkers

Cons

Poor work life balance

Micromanagement and excessive tracking

Too much pressure and workload

Salary Ranges

6 data points

L2

L3

L4

L5

L6

M3

M4

M5

M6

L2 · Data Scientist L2

0 reports

$104,000

total / year

Base

$41,600

Stock

$52,000

Bonus

$10,400

$72,800

$135,200

Interview Experience

5 interviews

Difficulty

3.6

/ 5

Duration

14-28 weeks

Offer Rate

60%

Experience

Positive 20%

Neutral 20%

Negative 60%

Interview Process

1

Application Review

2

Recruiter Screen

3

Technical Phone Screen

4

Technical Interview

5

Hiring Manager Interview

6

Offer

Common Questions

Coding/Algorithm

Technical Knowledge

Behavioral/STAR

Past Experience

System Design