Senior AI/ML Software Engineer

Job Title: Senior AI/ML Software Engineer Company: AMD Job Category: Machine Learning Roles & Responsibilities: 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. Responsibilities: 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. Work Location: Karnataka, India Availability: Full time

ACADEMIC CREDENTIALS:* Master’s or equivalent experience in Computer Science, Computer Engineering, or related field Qualifications: 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. Key Skills: Machine Learning, Python, TensorFlow, PyTorch Salary: Not disclosed