Senior AI/ML Software Engineer

Advanced Micro Devices, Inc.; Bengaluru Urban; On-site; INR 4,000,000 - 7,000,000; Full time

Job summary

Advanced Micro Devices, Inc. (AMD) seeks a Senior AI/ML Software Engineer to join the Machine Learning and Frameworks organization.

You will build and maintain tools and infrastructure to automate functional and performance validation of deep learning libraries and models across ROCm and GPU platforms. The role emphasizes Python and Linux-based systems, quality assurance, benchmarking and automation in the AI/ML space, with opportunities to work with the latest hardware and collaborate across

Qualifications

• Master's degree or equivalent in Computer Science, Computer Engineering, or related field.
• Strong Python development skills, with experience in test automation, CI/CD, and Linux scripting.
• Experience with machine learning frameworks (e.g., PyTorch, TensorFlow), model benchmarking, and ML model lifecycles.

Responsibilities

• Proficient in Python with performance tuning, debugging, and robust test design to ensure high-quality code.
• Design, implement, and manage CI/CD pipelines to enable rapid and reliable software deployment.
• Integrate automated testing frameworks into CI pipelines to ensure code quality, stability, and performance.

Skills

Python

Linux scripting

CI/CD

Test automation

Profiling

Education

Master's degree in Computer Science or related field

Tools

GitHub Actions

Buildkite

SLURM

Kubernetes

Job description

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

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 orientati