MTS - Deep / Machine Learning Software Engineer

Strong technical and analytical skills in C++ / Python AI development in a Linux environment. Ability to work as part of a team, while also being able to work independently, define goals and scope and lead your own development effort.

KEY RESPONSIBILITIES :

• Enable and optimize key AI models (LLM, Vision, MultiModal, etc.) on AMD GPUs
• Optimize AI frameworks like PyTorch, TensorFlow, etc. on AMD GPUs in upstream open-source repositories
• Collaborate and interact with internal GPU library teams to analyze and optimize training and inference for AI
• Work with open-source framework maintainers to understand their requirements and have your code changes integrated upstream
• Optimize GPU kernels, understand and drive AI operator performance (GEMM, Attention, etc.) with specialized teams
• Work in a distributed computing setting to optimize for both scale-up (multi-GPU) and scale-out (multi-node) systems
• Apply your knowledge of software engineering best practices

PREFERRED EXPERIENCE :

ACADEMIC CREDENTIALS :

Skills Required

Software Design, Machine Learning, C++, Linux, Debugging, Python