Senior AI/ML Software Engineer

Advanced Micro Devices, Inc; Bengaluru; On-site; INR 1,500,000 - 2,100,000; Full time

Job summary

Advanced Micro Devices, Inc. (AMD) in Bengaluru is seeking a Senior AI/ML Software Engineer to develop and maintain tooling and infrastructure for validating deep learning libraries and models across ROCm GPUs.

You will work on performance dashboards, automation, and scalable pipelines within the Machine Learning and Frameworks Org. The role emphasizes Python proficiency, Linux expertise, and collaboration with teams to enable reproducible model performance across hardware generations.

Qualifications

• Strong Python development skills with test automation and Linux scripting.
• Experience with ML frameworks (PyTorch, TensorFlow) and benchmarking workflows.
• Familiarity with CI/CD and performance dashboards.

Responsibilities

• Proficiency in Python for high-performance code, debugging, and reliable test design.
• Design, implement, and manage CI/CD pipelines (GitHub Actions/Buildkite) to enable rapid deployments.
• Integrate automated testing frameworks into CI pipelines for quality and reliability.
• Automate functional and performance testing of AI libraries and models across ROCm hardware.
• Develop tools for benchmarking and regression tracking across hardware generations and ROCm releases.
• Build real-time dashboards reporting performance and accuracy metrics for internal/public users.
• Collaborate with teams to support a range of models, including NDA workloads.
• Contribute to portable Python interfaces for multi-node profiling and containerized deployments.
• Support public AMD GitHub repositories and Docker releases for community use.

Skills

Python

CI/CD

Linux scripting

Performance profiling

Test automation

GitHub

Education

Master’s in Computer Science

Tools

SLURM

Kubernetes

GitHub

Buildkite

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believetechnology 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 AMDis shapingthefuture.

Whetheryou’redesigning next-gen processors, enabling AI breakthroughs, orbringing leading edge products to market, every role at AMD contributes to something bigger— technologythat 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 andFrameworks Org.As part of this team,you’llbuild andmaintaintools and infrastructure that automate functional and performance validation of deep learninglibraries &models acrossROCmand GPU platforms. Your contributions will directlyimpactdeveloper confidence, model portability, and transparent benchmarking for internal teams and the open-source community.You will be a member of a core team ofincredibly talentedindustry specialists and will work with the very latest hardware and software technology

THE PERSON:

We are seeking a software developer with strong technicalexpertisein 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,demonstratesexcellent 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:Proficiencyin 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 usingGitHub 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 andROCmreleases.
• Dashboard & Metrics Development:Build andmaintainreal-time dashboards that report relevant performance, accuracy, and reliability metrics for both internal and public users.
• Ecosystem Integration:Collaborate with teams likeDeepLearningModels(DLM) andMADengineto support a wide range of models, including public and private/NDA workloads.
• Client Enablement:Ensure out-of-box confidence forROCmclients by validatingmodel 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-facingAMDGitHub repositories and Docker releases, enabling the community to run andvalidatemodels onROCm.

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:Proficiencywith 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’sor equivalent experience in Computer Science, Computer Engineering, or related field

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.