AI/ML Compiler Engineer

Join AMD in Hyderabad/Secunderabad, Telangana, India, as a Software Development Engineer 2, contributing to the advancement of AI acceleration. You will be a key member of our core team, developing ML tools and methodologies to optimize and maximize system performance for AI workloads on Ryzen AI SoCs. This role involves working with the latest AI models in vision, language, and generative AI, collaborating with leading engineers from AMD's CPU, GPU, and Adaptable Compute teams.

Key Responsibilities:

• Contribute to the development and maintenance of AMD's AI compilers and runtime environments.
• Continuously research and implement performance-enhancing solutions.
• Stay abreast of software, hardware, ML algorithm, and architecture trends.
• Optimize existing systems and explore more efficient alternative approaches.
• Foster technical relationships with peers and partners.
• Collaborate with AMD architecture specialists to influence future product development.

Preferred Experience:

• Strong object-oriented programming background, with C/C++ preferred and at least 2 years of industry experience.
• Solid understanding of object-oriented design principles.
• Proficiency in software engineering principles, data structures, algorithms, operating systems, and multithreaded programming.
• Demonstrated ability to write high-quality, detailed code.
• Knowledge of AI frameworks such as PyTorch, TensorFlow, ONNXRuntime, TVM, and MLIR.
• Familiarity with key ML operations including GEMM and CONV.
• Understanding of AI acceleration performance implications across various compute, memory, communication configurations, and hardware/software implementation choices.
• Experience with software development processes and tools like debuggers, source code control (GitHub), and profilers.
• Effective communication and problem-solving skills are essential.

Academic Credentials:

• Bachelor's or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or a related field.