AI/ML Engineer

Shape the Future of Computing with AMD: Accelerating Innovation in AI and Beyond

At AMD, our mission is to develop exceptional products that accelerate next-generation computing experiences across AI, data centers, PCs, gaming, and embedded systems. We foster a culture of innovation and collaboration, believing that bold ideas, human ingenuity, and a shared passion for creating something extraordinary drive real progress. Joining AMD means discovering a culture where the real differentiator lies in our collective drive to push the limits of innovation and solve the world's most significant challenges. We strive for execution excellence while embodying directness, humility, collaboration, and inclusivity of diverse perspectives.

Join us as a Software Development Engineer 2 and advance your career.

The Role: Enabling AI Acceleration at Scale in Hyderabad

AMD is seeking an influential Software Engineer to play a pivotal role in enabling AI acceleration at scale. As a member of our core team, you will contribute to developing ML tools and methodologies to optimize and realize full system performance for AI workloads on Ryzen AI SoCs. You will work with the latest AI models addressing vision, language, and generative models, 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 innovative methods to enhance the performance of our AI solutions.
• Stay abreast of current software and hardware trends and innovations, with a specific focus on ML algorithms and architectures.
• Optimize existing systems and explore alternative, more efficient approaches to achieve project goals.
• Foster strong technical relationships with peers and external partners.
• Collaborate with AMD's architecture specialists to influence and improve future product designs.

Preferred Experience:

• A strong foundation in object-oriented programming, preferably with C/C++, and a minimum of 2 years of industry experience.
• A solid understanding of object-oriented design principles.
• Proficiency in software engineering principles, data structures, algorithms, operating system concepts, and multithreaded programming.
• Demonstrated ability to write high-quality code with exceptional attention to detail.
• Familiarity with key AI frameworks such as PyTorch, TensorFlow, ONNXRuntime, TVM, and MLIR.
• Experience with fundamental ML operations like GEMM and CONV.
• An understanding of the performance implications of different compute, memory, and communication configurations, as well as hardware and software implementation choices for AI acceleration.
• Experience utilizing software development processes and tools, including debuggers, source code control systems (e.g., GitHub), and profilers.
• Effective communication and problem-solving skills.

Academic Credentials:

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

AMD offers a comprehensive benefits package. AMD is an equal opportunity, inclusive employer committed to diversity and inclusion. We encourage applications from all qualified candidates and will provide accommodations as needed throughout the recruitment process. AMD may utilize Artificial Intelligence in its screening and selection processes, in line with our Responsible AI Policy.