ML Software Development Engineer

AMD AMD · Semiconductors · Hyderabad, India · Engineering

This role focuses on building and verifying ML-based software, specifically improving training efficiency and creating test cases for ML compilers. The engineer will work with C++, Python, and GPU/CPU communication, contributing to the AMD AI platform.

What you'd actually do

  1. Enabling support for different data type in hipblaslt/tensilelite
  2. Creating test cases to validate functionality and performance.
  3. Monitor and maintain end-to-end ML test cases for compiler in production, troubleshooting issues where needed
  4. Contribute your changes to open source.
  5. Influence the direction of AMD AI platform.

Skills

Required

  • C++
  • python
  • understanding of GPU/CPU communication
  • open-source code version control like Git

Nice to have

  • 5+ years of experience
  • debugging
  • profiling
  • perf analysis
  • HIP programming
  • ML pipeline
  • CI/CD pipeline
  • Strong communication and problem-solving skills

What the JD emphasized

  • defect free software for machine learning
  • improving training efficiency
  • innovating and coming up with new ideas

Other signals

  • ML based software
  • improving training efficiency
  • ML test cases for compiler