Lead Machine Learning Engineer

ADVANCE YOUR CAREER. ADVANCE THE WORLD.
At AMD, we believe technology can change lives for the better. It can heal us, entertain us, and make us more connected, productive, and understanding of the world around us. And were looking for talent who feel the same: people who want to leave the planet better than they found it, those who dont shy away from humanitys challenges but are determined to help solve them.
AMD is powering the next generation of supercomputing, high-performance computing, cloud, and AI. Whether youre designing next-gen processors, enabling AI breakthroughs, or creating go-to-market plans, every role at AMD contributes to something bigger — technology that moves the world forward.
Lead Machine Learning Engineer
AMD is seeking talented Machine Learning Engineers to join our AI/ML Solutions Team in Hyderabad. In this role, you will help deliver end-to-end AI/ML solutions by working with state-of-the-art neural network architectures—including Vision(CNN), Vision Transformers (ViTs), LLMs, BERT, Transformers, and custom models—and optimizing them for AMDs next-generation compute platforms.
You will be involved in compiler development, performance analysis, compiler optimization, software framework development, and system-level engineering. Working in a highly collaborative R&D environment, you will partner with hardware and software engineering teams, application experts, and customers to enable high-performance AI solutions across a range of AMD technologies.
This is an excellent opportunity to contribute to cutting-edge AI innovation while growing your expertise in machine learning, systems engineering, and product development.
Key Responsibilities

• Work with state-of-the-art AI/ML compiler toolchains across front-end, back-end, optimization, performance, and debugging domains.
• Collaborate with the AI/ML Solutions team to accelerate solution development and deployment.
• 
  Evaluate and optimize industry-standard and custom neural network architectures, including Vision(CNNs), Vision Transformers (ViTs), and Large Language Models (LLMs), for AMD platforms.
• Develop software frameworks, applications, and system-level solutions for machine learning workloads.
• Perform performance analysis, simulations, profiling, and optimization of AI applications.
• Contribute to a fast-paced, collaborative engineering environment focused on solving complex machine learning challenges.
  

Required Qualifications

• Bachelors or Masters degree in Computer Science, Computer Engineering, Data Science, Artificial Intelligence, or a related field.
• Experience with machine learning algorithms and deep learning frameworks.
• Strong understanding of deep neural network architectures such as CNNs, Vision Transformers (ViTs), and Transformer-based models.
• Hands-on experience with AI frameworks including: ONNX, PyTorch, TensorFlow, Caffe etc.
• Experience working with popular machine learning models such as: ResNet, YOLO, Vision Transformer (ViT), BEVFormer, Transformer-based models
• Excellent C/C++/Python programming and software design skills, including debugging, performance analysis, and test design.
  

Preferred Qualifications

• Experience collaborating with software partners on, Performance optimization, Benchmark development, Technical collateral creation (presentations, whitepapers, etc.), Joint demonstrations and proof-of-concept solutions
• Experience working directly with customers to understand and address critical design requirements.
• Ability to drive adoption of AI/ML compiler technologies through close engagement with customers and cross-functional teams.
• Experience collaborating with Product Marketing, Applications Engineering, and Architecture teams to define and architect new features.
• Hands-on experience with: Model Pruning, quantization, Model optimization, Performance tuning, Compiler technologies
• Knowledge of CUDA/HIP/OpenCL computing
• Experiences to optimize NPU/GPU/TPU kernels for performance is a plus
• Exposure to next-generation deep learning compiler technologies.
  

Benefits offered are described: AMD benefits at a glance.
AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee-based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third-party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants needs under the respective laws throughout all stages of the recruitment and selection process.
AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMDs Responsible AI Policy is available here.
This posting is for an existing vacancy.