
Senior ML Compiler Engineer
- Hyderabad, Telangana
- Permanent
- Full-time
- Design and Develop ML Compilers
- Lead the development of machine learning compilers specifically designed for AMD Neural Processing Units (NPUs).
- Debug and fix system level issues that occur while running ML models on AMD NPUs.
- Collaborate with cross-functional teams to seamlessly integrate machine learning solutions into existing workflows and systems.
- Performance optimization
- Optimize the ML compiler to improve the performance of machine learning models on the Neural Processing Unit (NPU).
- Propose ideas to further enhance efficiency and model execution speed.
- Project execution and delivery
- Ensure high-quality and timely project delivery, along with leading small teams.
- Break down complex features into manageable tasks to facilitate development and tracking and develop test plans for new features to ensure quality and coverage.
- Guide and support team members in achieving their project goals while facilitating their professional growth
- Proven ability to effectively solve complex problems.
- Hands on experience in debugging system level issues across the stack on SoCs, NPUs, etc.
- Knowledgeable in machine learning models & operators and data movement (tiling) strategies
- Familiar with software development tools and processes, including debuggers, source code control systems like GitHub, and profilers.
- Proficiency in object-oriented programming, with a strong preference for C/C++ languages.
- Experienced in Linux and Windows development environments
- Excellent communication skills, with a proven ability to motivate and lead teams effectively
- Knowledge in python, shell scripting, etc. is desirable
- Bachelor’s or Master's degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent