AMD is hiring for MTS Software Development Eng | Apply Now
Role :
MTS Software Development Eng
Location :
Bangalore, India
JOB DESCRIPTION:
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.
Explore more new job openings
What you will do
- 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
Education Required:
- Bachelor’s, Master’s or PhD degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent
skills
- 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
Get instant updates on the latest jobs! Join our WhatsApp and Telegram groups.
To join our WhatsApp channel and Telegram Group – Click on the WhatsApp and telegram icons below
