Senior Machine Learning Engineer, AI Solutions

Key Skills

Aartificial intelligence, Deep Learning, JAX framework, Machine learning techniques, PyTorch, TensorFlow

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.

THE PERSON:

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…

KEY RESPONSIBILITIES:

• 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, Collasal.ai)

PREFERRED EXPERIENCE:

• 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

ACADEMIC CREDENTIALS:

• 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

LOCATION:

Santa Clara, CA

LI-RL1

LI-Hybrid

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.