06 Sep
|
Advanced Micro Devices
|
Hyderabad
06 Sep
Advanced Micro Devices
Hyderabad
ADVANCE YOUR CAREER. ADVANCE THE WORLD.
At AMD, we believe technology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future.
Whether you’re designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger — technology that moves the world forward. Join us and, together, we’ll advance your career.
MTS SOFTWARE DEVELOPMENT ENGINEER
THE ROLE:
We are looking for a dynamic, energetic Lead / Staff Software Engineer to join our growing team in AI (Artificial Intelligence) group. In this role, the individual will be responsible for developing AI/ML specific C/C++ kernels and dataflow schedules for AMD Ryzen Processors built on XDNA Neural Processor Units (NPU) to map LLMs, Reliable Diffusion networks on NPU. As a C++ Kernel Developer, you will play a crucial role in designing, optimizing,
and implementing machine learning kernels specifically tailored for vector processors. Your work will directly impact the efficiency, speed, and accuracy of our machine learning models.
KEY RESPONSIBILITIES:
Kernel Development:
Design and implement highly optimized C++ kernel library for NPU/GPU.
Collaborate with the research and software teams to integrate these kernels into the existing software stack.
Vector Processor Optimization:
Work closely with hardware engineers to understand the architecture of VLIW vector core units such as MAC, GeMM, and non-linear functions.
Develop vectorized code that leverages SIMD (Single Instruction, Multiple Data) and ILP (instruction level parallelism) for maximum performance.
Performance Profiling and Tuning:
Profile and analyze the performance of existing kernels.
Identify bottlenecks and optimize critical sections for better throughput.
Testing and Validation:
Develop CPU models for the ML operators in C++/
📌 AI/ML Compiler Developer (Hyderabad)
🏢 Advanced Micro Devices
📍 Hyderabad