senior Machine Learning Engineer tech_leadership Master's Hybrid · Posted Feb 16, 2026

About this role

AMD is hiring a senior-level Machine Learning Engineer based in San Jose, CA (hybrid). The posting calls out experience with Python, TensorFlow, PyTorch, LLMs. Listed education preference: a master's degree or equivalent.

Role
Machine Learning Engineer
Function
machine learning
Level
senior
Track
Tech leadership
Location
San Jose, CA
Work mode
Hybrid
Education
Master's degree
Department
Engineering
Posted
Feb 16, 2026
AI Summary
Senior Machine Learning Engineer optimizing distributed training of large models on AMD GPUs at scale. Requires expertise in distributed training algorithms, ML frameworks (PyTorch, JAX, TensorFlow), and large-scale GPU cluster optimization. Focus on improving training efficiency, pipeline performance, and debuggability while contributing to open-source AI infrastructure.

More roles at AMD

Post‑Silicon SoC & System‑Level Validation Engineer
Bangalore, India · mid
Linux Embedded Systems Testing
SoC Validation Architect
Bangalore, India · mid
Embedded Systems Testing Observability
Software Development Engineer
Secaucus, NJ · mid
Python Bash Docker
Senior Analog Designer
Hyderabad, India · senior
Backend Development Embedded Systems
SRAM Circuit Design/ Memory Compiler Design Engineer
Hyderabad, India · mid
Testing Embedded Systems
All AMD jobs →

Job description

from AMD careers


WHAT YOU DO AT AMD CHANGES EVERYTHING 

At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you’ll discover the real differentiator is our culture. We push the limits of innovation to solve the world’s most important challenges—striving for execution excellence, while being direct, humble, collaborative, and inclusive of diverse perspectives. Join us as we shape the future of AI and beyond.  Together, we advance your career.  




 

THE ROLE:

We are looking for a Fellow/Sr. Fellow Machine Learning Engineer to join our Training At Scale team. If you are excited by the challenge of distributed training of large models on a large number of GPUs, and if you are passionate about improving training efficiency while innovating and generating new ideas, then this role is for you.  You will be part of a world class team focused on addressing the challenge of training generative AI.

 

THE PERSON:

The ideal candidate should have experience with distributed training pipelines, be knowledgeable in distributed training algorithms (Data Parallel, Tensor Parallel, Pipeline Parallel, Expert Parallel), and be familiar with training large models.

This is an excerpt. Read the full job description on AMD careers →
All machine learning jobs machine learning in San Jose, CA Jobs in San Jose, CA machine learning salaries machine learning career path
All AMD Jobs Browse machine learning roles senior positions