20 hours ago
Bengaluru, IndiaSenior
Responsibilities
- Develop and maintain Python tools and infrastructure for functional and performance validation of deep learning libraries and models.
- Design and manage CI/CD pipelines using GitHub Actions and Buildkite.
- Integrate automated testing frameworks into CI pipelines to assess code quality, stability, and performance.
- Automate model testing and validation across ROCm-supported AMD and NVIDIA hardware.
- Build benchmarking infrastructure for regression tracking across hardware generations and ROCm releases.
- Develop dashboards reporting performance, accuracy, and reliability metrics.
- Create portable Python interfaces supporting multi-node profiling, distributed workloads, and containerized deployments.
- Collaborate with DeepLearningModels and MADengine teams on public and private model workloads.
- Validate model performance and functionality in standardized, reproducible environments for ROCm clients.
- Support AMD GitHub repositories and Docker releases for open-source model validation.
Requirements
- Strong Python development skills with experience in performance tuning, debugging, robust test design, test automation, CI/CD, and Linux scripting.
- Familiarity with AI frameworks such as PyTorch and TensorFlow, model benchmarking, and machine learning model lifecycles.
- Familiarity with clusters and orchestration software such as SLURM and K8s.
- Experience with GitHub, Jenkins or similar tools, version control, testing strategies, code reviews, and collaborative software development.
- Strong experience with profiling tools, system monitoring, or regression tracking systems for deep learning models.
- Strong written and verbal communication skills and a proactive approach to defining and driving development efforts.
- Master’s degree or equivalent experience in Computer Science, Computer Engineering, or a related field.
Tech Stack
Categories
About AMD
AMD designs and sells CPUs, GPUs, and adaptive/embedded computing products for PCs, data centers, gaming, and edge devices. Its portfolio includes Ryzen and EPYC processors, Radeon and Instinct graphics, and adaptive SoCs from its Xilinx acquisition, sold to OEMs, cloud providers, and device makers. Founded in 1969 and headquartered in Santa Clara, it is a public company on NASDAQ and supplies semi-custom chips for major game consoles.