A

AMD

Designs chips for computers and AI systems

Lead AI Power & Performance Engineer

Applied Machine LearningPosted on Nov 5
Full-timeBengaluru, Karnataka, IndiaNo salary listedIn-person

About the Job

Requirements

  • 10 to 14+ years of relevant experience in AI performance engineering, system performance analysis, workload optimization, benchmarking, or post-silicon performance validation.
  • Hands-on experience evaluating, benchmarking, and characterizing AI inference workloads on CPU, GPU, APU, NPU, or heterogeneous computing platforms.
  • Strong experience with PyTorch, ONNX Runtime, vLLM, llama.cpp, TensorFlow, or equivalent AI frameworks and inference runtimes.
  • Deep expertise in platform architecture, hardware performance, and system-level optimization, including CPU, GPU, NPU, memory, power, and thermal subsystems.
  • Strong Linux systems expertise, including performance profiling, workload tracing, memory management, workload scheduling, automation, and system-level debugging.
  • Experience with profiling tools such as Linux perf, AMD uProf, ROCm Profiler, Nsight Systems, VTune, Perfetto, or equivalent tools.
  • Strong understanding of hardware architecture, memory hierarchy, data movement, power and thermal behavior, and performance trade-offs.
  • Experience with AI optimization techniques such as quantization, mixed-precision inference, graph optimization, operator fusion, kernel tuning, and memory optimization.
  • Proficiency in Python and Bash for benchmark automation, data analysis, and reporting.
  • Working knowledge of C or C++ for performance debugging and low-level workload analysis.
  • Ability to independently drive benchmarking strategy, competitive analysis, root-cause investigation, and technical recommendations.
  • Ability to convert complex performance data into clear technical reports, dashboards, presentations, and actionable recommendations.
  • Proven ability to provide technical leadership and drive complex initiatives across multiple engineering teams.

Responsibilities

  • Define and drive the AI performance benchmarking and workload characterization strategy across CPU, GPU, APU, NPU, and heterogeneous platforms.
  • Benchmark AI inference workloads using PyTorch, ONNX Runtime, vLLM, llama.cpp, TensorFlow, and related frameworks and runtimes.
  • Analyze performance metrics, including latency, throughput, time to first token, tokens per second, memory utilization, power consumption, and performance per watt.
  • Profile workloads and identify bottlenecks across AI frameworks, runtimes, libraries, drivers, operating systems, firmware, memory subsystems, and hardware.
  • Lead complex root-cause analysis and recommend optimizations related to quantization, graph execution, kernels, memory usage, workload scheduling, and heterogeneous compute utilization.
  • Conduct competitive benchmarking across AMD and competing CPU, GPU, APU, and AI accelerator platforms.
  • Collaborate with SoC architects and framework, runtime, compiler, driver, firmware, and system software teams to improve AI workload performance.
  • Develop and improve automation for benchmark execution, data collection, regression analysis, and results visualization.
  • Translate benchmarking data into actionable technical recommendations that influence software optimization, hardware design, product strategy, and customer engagements.
  • Present technical findings and recommendations to engineering teams, customers, executives, and other cross-functional stakeholders.
  • Provide technical leadership, mentor engineers, and drive alignment across multiple teams.

Desired Qualifications

  • Experience with ROCm, Ryzen AI Software, Vitis AI, MIGraphX, or related AI technologies.
  • Experience with MLPerf Inference or comparable AI benchmarking methodologies.
  • Knowledge of INT8, INT4, FP16, BF16, and mixed-precision execution.
  • Experience with large language model inference, including batching, KV-cache management, model parallelism, and memory optimization.
  • Experience with Docker, virtualization, and containerized AI environments.
  • Exposure to post-silicon bring-up, early silicon testing, competitive benchmarking, or customer-facing performance analysis.
  • Experience influencing platform architecture, software roadmaps, product strategy, or customer solutions through data-driven performance analysis.
  • Excellent executive, customer-facing, and cross-functional communication skills.

About the Company

AMD

Designs chips for computers and AI systems

AMD designs processors, graphics chips and AI accelerators for computer makers, cloud operators and businesses that build electronic devices. Its Ryzen processors power personal computers, while EPYC processors handle work in data centers and Radeon chips render graphics. Instinct accelerators run AI models, paired with software that developers use to train models and put them into production. AMD sells chips and related computing systems through equipment makers, distributors and retail channels, and contracts with outside foundries to fabricate its semiconductor designs. It was founded in 1969 and has its headquarters in Santa Clara, California.

Company size
10,001+
Company stage
IPO
Headquarters
Santa Clara, CA
Founded
1969

Recently Posted Jobs

Explore more opportunities at AMD

See more jobs