AMD - United States, San Jose, IT - Software Development

Join our Models and Applications team as a Principal ML Engineer. Optimize large-scale distributed training of generative AI models on AMD GPUs using PyTorch/JAX. Drive performance improvements in the end-to-end pipeline and contribute to open source. This role is based in San Jose or Bellevue, USA.