Senior ML Compiler Engineer

AMD AMD · Semiconductors · Hyderabad, India · Engineering

Senior ML Compiler Engineer role focused on designing, developing, and optimizing ML compilers for AMD NPUs, debugging system-level issues, and ensuring performance improvements for ML models.

What you'd actually do

  1. Lead the development of machine learning compilers specifically designed for AMD Neural Processing Units (NPUs).
  2. Debug and fix system level issues that occur while running ML models on AMD NPUs.
  3. Optimize the ML compiler to improve the performance of machine learning models on the Neural Processing Unit (NPU).
  4. Ensure high-quality and timely project delivery, along with leading small teams.

Skills

Required

  • C/C++
  • Linux
  • debugging system level issues
  • machine learning models & operators
  • performance optimization

Nice to have

  • Python
  • shell scripting
  • GitHub
  • profilers

What the JD emphasized

  • improving the performance of key applications and benchmarks
  • debug system level issues
  • machine learning models & operators

Other signals

  • ML compilers
  • AMD Neural Processing Units (NPUs)
  • performance optimization
  • debug system level issues