What is a Machine Learning Engineer at AMD?
At AMD, the Machine Learning Engineer role is distinct from typical data science positions at software-first companies. Here, you sit at the critical intersection of advanced software and cutting-edge hardware. You are not just building models; you are defining how the next generation of AI workloads runs on AMD’s Instinct accelerators, Ryzen AI processors, and the ROCm open software platform. This role is fundamental to AMD’s strategy of challenging the status quo in high-performance computing and generative AI.
You will work on enabling and optimizing deep learning frameworks (like PyTorch, TensorFlow, and JAX) to extract maximum performance from AMD silicon. This often involves diving deep into kernel optimization, compiler technologies, and distributed training strategies. Your work directly impacts how fast large language models (LLMs) train and infer, influencing the product capabilities of major cloud providers and enterprise partners who rely on AMD infrastructure.
This position offers a unique opportunity to work "under the hood" of modern AI. You will tackle complex challenges related to memory bandwidth, compute efficiency, and parallel processing. If you are passionate about squeezing every ounce of performance out of hardware and understanding the full stack—from the transistor to the transformer—this role is designed for you.
