A

AMD

Designs chips for computers and AI systems

AI Power & Performance Engineer

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeBengaluru, Karnataka, IndiaNo salary listedIn-person

About the Job

Requirements

  • Five to ten years of relevant experience in AI performance engineering, system performance analysis, workload optimization, or benchmarking.
  • Hands-on experience benchmarking AI inference workloads on CPU, GPU, APU, or NPU platforms.
  • Strong experience with AI frameworks or runtimes such as PyTorch, ONNX Runtime, vLLM, llama.cpp, or TensorFlow.
  • Strong Linux expertise, including performance profiling, workload tracing, automation, and system-level debugging.
  • Experience with profiling tools such as Linux perf, AMD uProf, ROCm Profiler, Nsight Systems, VTune, Perfetto, or equivalent tools.
  • Good understanding of CPU/GPU architecture, memory hierarchy, power and thermal behavior, and performance trade-offs.
  • Proficiency in Python and Bash for automation, data analysis, and reporting.
  • Ability to communicate complex performance data through technical reports, dashboards, and recommendations.
  • Bachelor’s or master’s degree in Computer Engineering, Electrical Engineering, Computer Science, Artificial Intelligence, Machine Learning, or a related field.

Responsibilities

  • Benchmark and characterize AI inference workloads using PyTorch, ONNX Runtime, vLLM, llama.cpp, TensorFlow, and related technologies.
  • Analyze latency, throughput, memory utilization, power consumption, and performance per watt.
  • Profile workloads and identify bottlenecks across AI frameworks, runtimes, operating systems, memory subsystems, and hardware.
  • Optimize workload performance through quantization, graph optimization, kernel tuning, memory optimization, and heterogeneous compute utilization.
  • Conduct competitive benchmarking across AMD and competing CPU, GPU, APU, and AI accelerator platforms.
  • Automate benchmark execution, data collection, regression analysis, and results visualization.
  • Convert benchmarking results into actionable recommendations for software, hardware, product, and business teams.
  • Collaborate with architects, framework, runtime, driver, firmware, and system software teams to resolve performance issues.

Desired Qualifications

  • Experience with ROCm, Ryzen AI Software, Vitis AI, MIGraphX, or related AI technologies.
  • Knowledge of model quantization, mixed-precision inference, graph optimization, and LLM-serving techniques.
  • Experience with MLPerf Inference, competitive benchmarking, Docker, virtualization, or post-silicon performance validation.

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