Senior AI/ML Software Engineer

ADVANCE YOUR CAREER. ADVANCE THE WORLD. 

At AMD, we believe technology has the power to solve the world’s most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future. 

Whether you’re designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger — technology that moves the world forward. Join us and, together, we’ll advance your career.

Senior AI/ML Software Engineer

 
THE ROLE: 

AMD is looking for a skilled and motivated software engineer to join the Machine Learning and Frameworks Org. As part of this team, you’ll build and maintain tools and infrastructure that automate functional and performance validation of deep learning libraries & models across ROCm and GPU platforms. Your contributions will directly impact developer confidence, model portability, and transparent benchmarking for internal teams and the open-source community. You will be a member of a core team of incredibly talented industry specialists and will work with the very latest hardware and software technology
 

THE PERSON: 

We are seeking a software developer with strong technical expertise in Python and Linux-based systems, who is passionate about quality assurance, benchmarking, and automation in the AI/ML space. The ideal candidate thrives in both collaborative and independent environments, demonstrates excellent problem-solving skills, and takes ownership in defining goals and delivering impactful solutions. Experience working with machine learning frameworks, performance dashboards, or automation platforms is a strong plus. 

KEY RESPONSIBILITIES 

• Software Engineering Excellence: Proficiency in Python with experience in performance tuning, debugging, and robust test design, ensuring reliable, maintainable, high-performance codebases.   
• CI/CD Pipeline Development: Design, implement, and manage efficient continuous integration and delivery pipelines using GitHub Actions and Buildkite to enable rapid and reliable software deployment.
• Quality Assurance Enablement: Integrate automated testing frameworks into CI pipelines to ensure code quality, stability, and performance across development cycles.
• Model Testing & Validation: Automate functional and performance testing of AI libraries & models across ROCm-supported hardware (AMD and NVIDIA) using scalable tools and pipelines. 
• Benchmarking Infrastructure: Develop tools for continuous benchmarking and regression tracking across hardware generations and ROCm releases. 
• Dashboard & Metrics Development: Build and maintain real-time dashboards that report relevant performance, accuracy, and reliability metrics for both internal and public users. 
• Ecosystem Integration: Collaborate with teams like DeepLearningModels (DLM) and MADengine to support a wide range of models, including public and private/NDA workloads. 
• Client Enablement: Ensure out-of-box confidence for ROCm clients by validating model performance and functionality in standardized and reproducible environments. 
• Scalable Tooling: Contribute to the design of portable, easy-to-use Python interfaces that support multi-node profiling, distributed workloads, and containerized deployments. 
• Open-Source Contributions: Support public-facing AMD GitHub repositories and Docker releases, enabling the community to run and validate models on ROCm. 

PREFERRED EXPERIENCE:  

• Programming & Tooling: Strong Python development skills, with experience in test automation, CI/CD, and Linux scripting. 

• Machine Learning Workflow Understanding: Familiarity with AI frameworks (e.g., PyTorch, TensorFlow), model benchmarking, and ML model lifecycles.  

• DevOps: Familiarity with clusters and orchestration software (SLURM, K8s). Experience with GitHub, Jenkins or similar and development workflows 

• Software Engineering Practices: Proficiency with version control (GitHub), testing strategies, code reviews, and collaborative software development. 
• Performance Analysis: Strong experience with profiling tools, system monitoring, or regression tracking systems for deep learning models.

• Communication & Ownership: Strong written and verbal communication skills with a proactive approach to defining and driving development efforts. 

ACADEMIC CREDENTIALS:  

• Master’s or equivalent experience in Computer Science, Computer Engineering, or related field

#LI-PK1

Benefits offered are described:  AMD benefits at a glance.

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.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position.  AMD’s “Responsible AI Policy” is available here.

This posting is for an existing vacancy.