MTS Software Development Eng. Job Listing at AMD in Bengaluru, KA (Job ID 40053-en-us)

Description

WHAT YOU DO AT AMD CHANGES EVERYTHING

We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellence while being direct, humble, collaborative, and inclusive of diverse perspectives. 

AMD together we advance_

MTS SOFTWARE DEVELOPMENT ENGINEER 

AMD Artificial Intelligence Group (AIG) leads AMD AI strategy and drives AI roadmap across client, edge, and cloud. We build AI capabilities, including silicon, software, models, use cases, to create a vibrant AMD AI ecosystem together with everyone.

Our organization, AIG SHARK, aims to build AI software solutions that are unified, performant, flexible, and customizable. We approach the mission with open-source and community-driven principles. We believe in the power of the community and together we can build a great stack that benefits everyone. Being a part of AMD and AIG, it also means we have a wide portfolio of hardware and customers to support for realized product excellence and business impacts. 

Our group maintains and contributes heavily to a number of open source projects. If you work in any of these areas and would like to consider full time opportunities with AMD in relation to these projects, please say so when you reach out. Even if we don't have a perfectly matching position open right now, we often consider candidates with a track record of collaboration and contributions.

• IREE
• Torch-MLIR
• LLVM / MLIR
• Turbine

The Role: 

Develop feature and improve quality of ONNX importer, ONNX Runtime Execution Provider, ONNX to LinAlg lowering, Torch MLIR to LinAlg lowering, and add support for new OPs and passes in LLVM/MLIR, torch MLIR and IREE compiler.

The Person
The ideal candidate should be passionate about implementing and improving effective algorithms and techniques to create and enhance compiler for machine learning and possess leadership skills to drive sophisticated issues to resolution. The candidate should be able to communicate effectively and work optimally with different teams across AMD.

Key Responsibilities

• Implement and improve support for new operators and passes in the compiler
• Mentor and provide guidance to others
• Learn latest industry trends and bring new ideas to the team
• Design and develop new groundbreaking AMD technologies
• Debugging/fix existing issues and research alternative, more efficient ways to accomplish the same work

Preferred Experience (post College)

• Strong object-oriented programming background, C/C++ and Python
• ML Compiler and Runtime technologies: MLIR, XLA, OpenXLA, IREE, OpenAI Triton compiler
• Compiler building skills
• Code generation for a ML hardware accelerator
• Understanding of PyTorch, Tensorflow, ONNX, JAX etc.
• Ability to write high quality code with a keen attention to detail
• Experience with modern concurrent programming and threading APIs
• Experience with software development processes and tools such as debuggers, source code control systems (GitHub) and profilers is a plus
• Effective communication and problem-solving skills

Academic Credentials

• Bachelor's, Master's or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent

#LI-VV1

Benefits offered are described:  AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law.   We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.