What are the responsibilities and job description for the AI/ML Compile Devolpement Engineer, Mid-Level position at Jobright.ai?
Verified Job On Employer Career Site
Job Summary:
Advanced Micro Devices, Inc (AMD) is a technology company focused on transforming lives with their products. They are seeking a Compiler Development Engineer to architect and define AI workload models and optimize performance for Neural Processing Units (NPU). The role involves collaboration with cross-functional teams to enhance AI model development and performance analysis.
Responsibilities:
• Research, design, and implement novel methods for efficient CNN, GEN AI models.
• Model optimization method design including quantization, sparsity, NAS, etc.
• Collaborate with other team members and teams.
• Collaborate with compiler team to develop optimization strategies for the compiler.
Qualifications:
Required:
• Strong engineering skills to tackle complex challenges on AI model development work.
• Experience in optimizing and accelerating CNN/Generative AI models.
• Excellent cross team collaboration skills.
• Strong experience in developing ML compiler for efficient network mapping on NPU.
• Ability to work with cross-functional teams to optimize various parts of the SW stack – AI Compiler, AI frameworks, device drivers, and firmware.
• Experience in bringing up emerging ML models based on CNN, transformers and characterizing performance.
• Experience in working on quantization, sparsity, and architecture search methods to optimize and enhance the performance, efficiency, and accuracy of Generative AI models.
• Ability to collaborate closely with software engineers, data scientists, and researchers to integrate AI models into software applications and platforms.
• Research, design, and implement novel methods for efficient CNN, GEN AI models.
• Model optimization method design including quantization, sparsity, NAS, etc.
• Collaborate with other team members and teams.
• Collaborate with compiler team to develop optimization strategies for the compiler.
• Minimum of a BS degree.
Preferred:
• Experience with deep learning framework, e.g., Pytorch/ONNX/TensorFlow.
• Experience on model compression, quantization, and end-to-end inference optimization.
• Strong coding skills in C/C , Python required.
• Experience with any of the following also a plus: LLMs, stable diffusion, NeRF, or text-to-video generation.
• Solid knowledge of AI and ML concepts and techniques.
• Practical experience applying these concepts to solve real-world problems in the context of research or work experience.
• Understanding the performance implications on AI acceleration of different compute, memory, and communication configurations and hardware and software implementation choices.
• Developing and optimizing code for VLIW processors.
• Analyzing code for high performance CONV, GEMM and non-linear operators.
• Deep understanding of AI frameworks, preferably ONNX.
• MS or above preferred.
Company:
Advanced Micro Devices is a semiconductor company that designs and develops graphics units, processors, and media solutions. Founded in 1969, the company is headquartered in Santa Clara, California, USA, with a team of 10001 employees. The company is currently Public Company.
