Senior Machine Learning Engineer, AI Solutions

Advanced Micro Devices, Inc.
Apply Now

Job Description

This role will be part of the team of Software Engineers that are designing the next generation of ML and AI into the Enterprise Data Center GPU space.


We are looking for a machine learning engineer that is experienced in running large, distributed models on training and inferencing clusters. If terms like Horovod, MoE, FSDP, DDP, ZerO and Parameter servers are second nature, you are the right person. The large the cluster, the better your fit…


  • Experience training models on large clusters and managing checkpointing, learning rates, data sharing etc.
  • Experience deploying inferencing models using tensor and model parallel techniques.
  • Play a role in all the phases of software & model development, from requirement gathering, analysis, design, development, testing and final release to customers.
  • Provide clear and timely communication related to status and other key aspects of the project to leadership team.
  • Responsible for working with data scientist teams for resolving critical customer escalations on a need basis.
  • Willingness to learn skills, tools, and methods to advance the quality, consistency, and timeliness of AMD software products.
  • Experience with deep learning models and associated AI frameworks (PyTorch, TensorFlow, Jax (FLAN), Lightning, Deepspeed, Mosaic, Horovod,


  • Deep experience with distributed training and inferencing frameworks like Horovod, Deepspeed, Lighting, Huggingface Training, Mosaic composer, Megatron
  • Deep experience with AI/ML frameworks like Pytorch, Tensor flow, JAX, etc. is preferred.
  • Knowledge of Acceleration platforms like GPU, TPU, APU, FPGAs in preferred


  • BS with several years of related experience or MS with years of related experience or PhD with years of related experience in Computer Engineering or Electrical Engineering or Computer Science or related equivalent from top universities

This is a senior level position, new grads need not apply


Santa Clara, CA



At AMD, your base pay is one part of your total rewards package. Your base pay will depend on where your skills, qualifications, experience, and location fit into the hiring range for the position. You may be eligible for incentives based upon your role such as either an annual bonus or sales incentive. Many AMD employees have the opportunity to own shares of AMD stock, as well as a discount when purchasing AMD stock if voluntarily participating in AMD’s Employee Stock Purchase Plan. You’ll also be eligible for competitive benefits described in more detail here.

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.

Company Info.

Advanced Micro Devices, Inc.

Advanced Micro Devices, Inc. (AMD) is an American multinational semiconductor company based in Santa Clara, California, that develops computer processors and related technologies for business and consumer markets. While it initially manufactured its own processors, the company later outsourced its manufacturing, a practice known as going fabless, after GlobalFoundries was spun off in 2009. AMD's main products include microprocessors, motherboard

  • Industry
    Artificial intelligence,Video games,Semiconductors,Computer hardware
  • No. of Employees
  • Location
    Santa Clara, CA, USA
  • Website
  • Jobs Posted

Get Similar Jobs In Your Inbox

Advanced Micro Devices, Inc. is currently hiring Senior Machine Learning Engineer Jobs in Santa Clara, CA, USA with average base salary of $178,560 - $267,840 / Year.

Similar Jobs View More