Machine Learning Stack Integration Eng. - 85062

Location: Santa Clara, California, US

Company: Advanced Micro Devices

Apply now

Apply for Job

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 Stack Integration Engineer

The Role:

AMD is looking for people who will be working with Solution Validation & Debug within the Machine Learning Software Engineering group. As a team member responsibility will include working closely on the debug and triage of Machine learning and High-Performance Computing related issues that add value to the Solution Validation of ROCm Stack.

The Person:

The ideal candidate will bring broad experience on dealing with complex software level issues related to Machine Learning and High-Performance Computing.

Key Responsibilities:

  • Debug Machine Learning/ High Performance Computing related issues on Radeon Open Compute Stack (ROCm)
  • Develop test contents for sophisticated Machine learning algorithms on distributed nodes
  • Port High Performance computing application on ROCm
  • Reproduce field defects and develop appropriate tests to prevent future issues.
  • Design, develop and deploy testing tools and automation libraries vital to perform testing.
  • Be responsible for the adoption of tooling and industry standard methodologies by means of advocacy and outreach to help our development communities’ level up.

Preferred Experience:

  • Languages: Python, C, C++, Linux Shell scripting.
  • Frameworks/Libraries: TensorFlow, PyTorch, ONNXRT
  • Tools: Prior experience with Linux, Docker, LLVM compilers
  • Understanding of High-Performance Computing application, Machine learning and GPU Programming, MPI Parallel Programming

Academic Credentials:

  • Bachelor's Degree in Computer Science or related quantitative field. An advanced degree or equivalent practical work experience is a plus.


Santa Clara, CA

Austin, TX



Requisition Number: 85062 
Country: United States State: California City: Santa Clara 
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

Apply for Job

Share this Job