GPU Performance Modeling & Optimization Engineer

AMD; Bengaluru Urban; On-site; INR 4,000,000 - 7,000,000; Full time

Benefits offered by this job

AMD benefits

Job summary

AMD in Bengaluru is seeking an experienced GPU Performance Modeling and Optimization Engineer to push pre-silicon performance modeling, feature exploration, and workload optimization. You will own post-silicon characterization and hardware–software correlation.

You will collaborate with compiler, runtime, and software teams to optimize graphics and AI workloads and bridge RTL design with software stacks. The role demands deep GPU architecture knowledge and hands-on modeling using C++ and Python.

Qualifications

• 5–10 years of industry experience in silicon performance engineering, GPU modeling, or microarchitecture design.
• Strong experience in pre-silicon performance modeling and hardware–software co-design.
• Proficiency in C++ and Python.
• Hands-on with Linux and lab-based silicon characterization.

Responsibilities

• Pre-silicon performance modeling and feature exploration using cycle-accurate simulators.
• Hardware–software co-design and workload optimization for AI and graphics workloads.
• Post-silicon characterization, correlation, and lab-based debugging.
• Collaborate with compiler, runtime, and software frameworks to optimize workloads.
• Bridge hardware design RTL and software stacks to translate workload requirements into constraints.

Skills

GPU modeling

C++

Python

LLM scaling

Data parallelism

Graphics pipeline

Performance profiling

Education

BE/B.Tech/M.Tech in relevant field

Tools

Linux tracing frameworks

PCIe interconnects

Job description

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.

The Role:

We are seeking an experienced, hands‑on GPU Performance Modeling and Optimization Engineer to join our core engineering team. Your primary focus will be pre‑silicon performance modeling, feature exploration, and workload optimization, coupled with owning post‑silicon characterization, hardware‑software correlation, and lab‑based performance debug.

You will be responsible for ensuring our upcoming SoCs/GPUs deliver best‑in‑class performance across both traditional graphics pipelines (compute shaders, ray tracing, rasterization) and cutting‑edge compute/AI workloads (Transformer‑based models, Large Language Models (LLMs), Vision models etc).

Key Responsibilities:

1. Pre‑Silicon Performance Modeling & Feature Exploration

• Simulator: Leverage and maintain highly accurate, modular cycle‑accurate or cycle‑approximate simulators for key GPU subsystems (e.g., Shader Engines, Cache Hierarchies, Memory Subsystems, and Interconnects).
• Microarchitectural Exploration: Define and execute rigorous simulation experiments to evaluate proposed GPU configurations, scaling limits, and trade‑offs. Provide data‑driven recommendations backed by thorough sensitivity analyses.
• Workload Characterization: Trace, analyze, and profile complex workloads to extract structural execution footprints. Translate these insights into microarchitectural bottlenecks and establish bounding box for performance for various workloads.

2. Hardware‑Software Co‑Design & Workload Optimization

• Compute & LLM Scaling Optimization: Profile and optimize performance for advanced generative AI and LLM topologies. Identify bottlenecks across the compute engine, local memory hierarchy (L1/L2), and SoC fabrics.
• Graphics Stack Optimization: Analyze execution efficiency across graphics shaders and compute‑heavy pipelines to maximize execution unit utilization and minimize latency.
• Cross-Layer Collaboration: Partner with compiler, runtime, and software framework teams (e.g., PyTorch, Triton) to implement/recommend optimizations.

3. Post‑Silicon Characterization, Correlation & Debug

• Pre-to-Post Correlation: Lead efforts to execute workloads on early silicon, capture performance telemetry, and systematically correlate results back to pre‑silicon performance models to improve simulator fidelity.
• Lab Debug & Telemetry: Root‑cause hardware‑software execution mismatches and unexpected performance drops on physical silicon using low‑level performance counters, register dumps, and Linux tracing frameworks.

Cross‑Functional Alignment:

• Act as the technical bridge between hardware design (RTL) and software stacks, translating high‑level workload requirements into clear hardware architectural constraints.

Preferred Experience & Skills:

• GPU Architecture Depth: Excellent technical knowledge of GPU execution pipelines, SIMD/SIMT models, cache hierarchy management, memory technologies, and high‑bandwidth interconnects (e.g., PCIe, custom fabrics).
• Advanced Performance Modeling: Strong, hands‑on experience building or significantly contributing to execution‑driven/cycle‑accurate hardware performance simulators. Expert proficiency in modern C++ and production‑grade Python.

• Workload Domain Expertise:
  • AI/Compute: Practical understanding of LLM scaling techniques, parallel training/inference strategies (Tensor, Pipeline, Data parallelism), and kernel execution graphs.
  • Graphics: Solid grounding in modern programmable graphics pipelines and compute shader mechanics.

• Profiling & Post‑Si Debugging: Proven capability to extract and analyze low‑level hardware performance counters, profile system workloads under Linux, and work within a lab environment for silicon characterization.

• Delivery & Execution: Excellent structured problem‑solving skills. Ability to translate raw, noisy simulation or silicon data into crisp, actionable technical findings for engineering stakeholders.

Academic Credentials & Experience Requirements:

• B.E. / B.Tech. / M.Tech. / with 5 to 10 years of industry experience in silicon performance engineering, GPU modeling, or microarchitecture design.

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.