Senior AI/ML Software Engineer
AMD · Posted Nov 5
High-performance processors, graphics cards, data center solutions, and AI accelerators
Get a personal compatibility score
Add a resume for personal matches
About the role
AMD's Machine Learning and Frameworks Org builds tools and infrastructure that automate functional and performance validation of deep learning libraries and models across ROCm and GPU platforms. The team focuses on developer confidence, model portability, and transparent benchmarking for internal teams and the open-source community. This senior software engineer will develop CI/CD pipelines, benchmarking infrastructure, dashboards, and open-source tooling for AI/ML workloads.
What you will do
- 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.
Skills used in this role
What the employer is looking for
- 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.
- Master’s or equivalent experience in Computer Science, Computer Engineering, or related field
Preferred qualifications
- 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.
Benefits and support
- Compensation and benefits are detailed in the job posting
About AMD
AMD is a global semiconductor company that designs high-performance central processing units (CPUs), graphics processing units (GPUs), and adaptive processors for data centers, personal computers, gaming consoles, and embedded systems. The company specializes in advanced computing solutions and AI accelerators, positioning itself as a major leader in modern high-performance and artificial intelligence hardware. Through strategic developments and acquisitions, AMD continues to expand its ecosystem across cloud infrastructure, enterprise computing, and physical AI.
- Industry
- Semiconductors
- Company size
- 31000 employees
- Founded
- 1969-05-01
- Location
- Santa Clara, California, USA
- Funding stage
- Public Company
Funding
Public Company
- 2007-11-16Post Ipo
Leadership
Chair and Chief Executive Officer
Executive Vice President, Chief Financial Officer and Treasurer
Executive Vice President and Chief Technology Officer
Executive Vice President and Chief Commercial Officer
Executive Vice President and General Manager, Data Center Solutions Business Group
Recent coverage
AMD Newsroom
AMD Reports Third Quarter 2026 Financial Results2026-11-03
PC Gamer
AMD says it's going to beat Intel in total revenue for the very first time by the end of 20262026-11-04
Shattered
AMD rides data center riches to 'largest and most profitable' quarter in history2026-11-04
Investing.com
AMD tops Q3 2026 estimates as stock jumps after hours on record data center growth2026-11-03