Machine Learning Performance Engineer.- 100521

Company: AMD
Location: Austin, Texas, United States
Type: Full-time
Posted: 03.SEP.2021
< >


What You Do At AMD Changes Everything At AMD, we push the boundaries of what is possible. We believe in changing the world for the better b...


What You Do At AMD Changes Everything

At AMD, we push the boundaries of what is possible. We believe in changing the world for the better by driving innovation in high-performance computing, graphics, and visualization technologies - building blocks for gaming, immersive platforms, and the data center.

Developing great technology takes more than talent: it takes amazing people who understand collaboration, respect, and who will go the "extra mile" to achieve unthinkable results. It takes people who have the passion and desire to disrupt the status quo, push boundaries, deliver innovation, and change the world. If you have this type of passion, we invite you to take a look at the opportunities available to come join our team.

Machine Learning Performance Engineer

The Role

This is a highly visible position with the opportunity to work closely with teams across AMD and impact MLPerf optimizations on AMD hardware and software architectures.

The Person

We are seeking a machine learning specialist with a background in frameworks such as PyTorch/TensorFlow, and machine learning workloads. The successful candidate will have a passion for machine learning algorithms and understand the mapping of the workloads (such as CNNs, Transformer, BERT, recommendation models) on CPU/GPU architecture.

Key Responsibilities

  • Define next generation algorithmic and software optimizations by conducting in-depth performance analysis of workloads and presenting the design and implementation choices.
  • Work with internal AMD teams and external customers to understand and develop software optimizations at machine learning frameworks level.
  • Drive development, product definitions, and customer interactions that lead to the development of AMD outstanding machine learning platforms
  • Develop machine learning software optimizations specific to AMD GPU platforms.
Preferred Experience
  • Expertise with PyTorch/TensorFlow and performance tools such as nvprof and rocprof.
  • Experience with standard machine learning models such as ResNet50, Transformer, SSD, BERT, DLRM and Reinforcement Learning
  • Familiarity with large scale execution of machine learning models on clusters connected with Ethernet/InfiniBand
Academic Credentials

Bachelor's or Master's in Computer Science, Computer Engineering, or related subjects, or equivalent experience.


Austin, TX

Requisition Number: 100521

Country: United States State: Texas City: Austin

Job Function: Design

AMD does not accept unsolicited resumes from headhunters, recruitment agencies or fee based recruitment services. AMD and its subsidiaries are equal opportunity employers. We consider candidates regardless of 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. Please click here for more information.

Apply Now


Loader2 Processing ...