Sr. AI/ML Software Engineer

Advanced Micro Devices (AMD); Bengaluru; On-site; INR 4,500,000 - 6,500,000; Full time

Job summary

AMD is seeking a Senior AI/ML Software Engineer to join the Machine Learning and Frameworks Org in Bengaluru. You will build and maintain tools and infrastructure that automate validation of deep learning libraries across ROCm and GPU platforms.

The role emphasizes Python/Linux, quality assurance, benchmarking, and automation, with opportunities to influence open-source projects and collaborate across teams.

Qualifications

• Master's degree in Computer Science or related field.
• Experience with AI frameworks (PyTorch, TensorFlow) is a plus.
• Strong Python and Linux scripting/automation experience.

Responsibilities

• Build and maintain tools and infrastructure for automated validation of deep learning libraries across ROCm and GPU platforms.
• Develop CI/CD pipelines using GitHub Actions and Buildkite.
• Integrate automated testing frameworks into CI pipelines to ensure code quality and performance.
• Automate functional and performance testing of AI models across ROCm-supported hardware.
• Develop tools for benchmarking and regression tracking across hardware generations and ROCm releases.
• Create real-time dashboards reporting performance, accuracy, and reliability metrics.
• Collaborate with teams like DeepLearningModels and MADengine to support diverse workloads.
• Validate model performance in reproducible environments for ROCm clients.
• Design portable Python interfaces supporting multi-node profiling, distributed workloads, and container deployments.
• Contribute to public AMD GitHub repositories and Docker releases.

Skills

Python

Linux

Test automation

CI/CD

Performance benchmarking

Education

Master's in Computer Science/Engineering

Tools

GitHub Actions

Buildkite

Docker

Kubernetes

Job description

Job Summary

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

Disclaimer: This job posting has been aggregated from external source. Role details, content, and availability are subject to change. Applicants are advised to confirm the latest information directly on the company website before applying.