Artificial Intelligence Performance Engineer

AMD - Santa Clara, CA

Full-time - Entry Level

Santa Clara, CA

Computer and Electronic Product Manufacturing

About the position

The Artificial Intelligence Performance Engineer at AMD is responsible for ensuring optimal performance of AMD Instinct GPU-accelerated systems before deployment. This role involves a deep understanding of GPU performance, performance monitoring, and tuning tools, alongside data science and communication skills to effectively share findings with engineering and business teams. The position emphasizes innovation and collaboration, aiming to enhance the performance and efficiency of AI and ML applications.

Responsibilities

• Define performance suite and best practices for measuring GPU-accelerated workloads to assess scalability and efficiency of AI models and algorithms
• Benchmark and analyze AI workloads in single and large multi-node configurations comparing against previous generations and competitors
• Perform comprehensive performance analysis and report findings for the entire platform including GPU, CPU, interconnects, network, software stack, etc.
• Identify performance bottlenecks that impact data center GPU-accelerated workloads, tune and collaborate with other software teams to improve performance
• Stay up to date with emerging technologies and trends in the AI field and explore ways to improve the performance of GPU-accelerated workloads at scale

Requirements

• Solid knowledge of Artificial Intelligence (AI) and Machine Learning (ML) concepts and techniques, including deep learning, reinforcement learning, natural language processing, generative AI, and computer vision
• Experience in benchmarking methodologies, performance analysis, workload profiling, performance monitoring and debugging tools
• Advanced Linux OS, container (e.g. Docker) and GitHub skills
• Programming skills in a variety of relevant languages such as Python or C/C++
• Expertise with deep learning frameworks like PyTorch and TensorFlow
• In-depth knowledge of GPU acceleration with either AMD or Nvidia GPU compute products
• Excellent problem-solving skills and an automation mindset

Nice-to-haves

• Knowledge and interest in computer and GPU architecture
• Inquiring mind and a passion for improving skills and driving performance

Benefits

• Base pay dependent on skills, qualifications, experience, and location
• Eligibility for annual bonus or sales incentive
• Opportunity to own shares of AMD stock
• Discount when purchasing AMD stock through Employee Stock Purchase Plan
• Competitive benefits package