Principal AI/ML SW/HW Co-Design Lead

Advanced Micro Devices, IncSanta Clara, CA
9d

About The Position

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. AMD is looking for a strategic AI/ML SW/HW CO-DESIGN LEAD who is passionate about improving the performance of key AI/ML applications and benchmarks. You will be a member of a core team of incredibly talented industry specialists and will work with the very latest hardware and software technology.

Requirements

  • Strong object-oriented programming background, proficiency Python and C/C++ preferred
  • Experience on GPU programming(HIP/CUDA/ROCm) & performance profiling/optimization preferred
  • Experience with modern concurrent programming and threading APIs on Windows/Linux operating system development
  • Direct experience with or deep exposure to GPU/TPU/CPU hardware, subsystems or core blocks design
  • Familiar with GPU/TPU/CPU SOC architectures including SIMD/SIMT/Parallel Processing models, Cache/ Memory hierarchy subsystem
  • Effective communication, issue analysis and problem-solving skills
  • Motivating leader with good interpersonal skills
  • Bachelor’s or Master's or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent

Nice To Haves

  • Familiar with training model & framework tunning is a plus
  • Knowledge of cluster and multi-node systems, AI/ML performance analysis, and SW simulation and emulation is a valuable asset

Responsibilities

  • Work with AMD’s architect specialists to influence & improve future product designs
  • Apply a data-driven and structured approach to issue analysis and optimization efforts
  • Contribute to AMD software and hardware innovation, algorithms and architectural design changes
  • Design and develop new groundbreaking AMD technologies
  • Drive and lead AI/ML software shift left efforts
  • Debug, analyze, resolve existing issues and explore more efficient alternatives, influence design enhancements for next-gen HW & SW.
  • Develop technical relationships with peers and partners

Benefits

  • AMD benefits at a glance.
© 2024 Teal Labs, Inc
Privacy PolicyTerms of Service