Senior AI Platform Engineer: Scalable ML Infrastructure

Advanced Micro Devices; Santa Clara (CA); On-site; USD 190,000 - 260,000; Full time

Job summary

Advanced Micro Devices in Santa Clara, CA is seeking AI/ML Platform Engineers to build a scalable platform layer for AI-for-engineering workflows. The role focuses on infrastructure that supports large-scale agent execution, distributed training and inference, experiment tracking, and artifact management.

You will collaborate with ML Systems Research Engineers, AI Researchers, and hardware domain experts to operationalize the Blueprint framework across kernel optimization, RTL/PPA, verification,

Qualifications

• Bachelor's degree in CS/CE/EE/ML or related field; Master’s preferred; PhD a plus.
• Strong programming skills in Python and at least one systems language (C++, Go, Rust).
• Experience with distributed ML platforms, GPU infrastructure, and production reliability.

Responsibilities

• Build and operate the shared AI platform for agentic engineering workflows including submission, scheduling, orchestration, retries, logging, artifact storage, and experiment tracking.
• Develop infrastructure for distributed training and inference across GPU clusters; batch evaluation and large-scale agent rollout.
• Build platform services for benchmark execution, correctness checking, profiling, regression tracking, and reproducible evaluation.
• Maintain artifact systems for kernels, traces, logs, and simulator outputs.
• Partner with ML Systems Research Engineers and Applied AI Engineers to productionize research workflows and reusable Blueprint harnesses.

Skills

Python

C++

Go

Rust

Education

Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, Machine Learning

Master's degree (preferred)

Tools

Kubernetes

Ray

Slurm

CI/CD

Job description

Advanced Micro Devices in Santa Clara, CA is seeking AI/ML Platform Engineers to build a scalable platform layer for AI-for-engineering workflows. The role focuses on infrastructure that supports large-scale agent execution, distributed training and inference, experiment tracking, and artifact management.

You will collaborate with ML Systems Research Engineers, AI Researchers, and hardware domain experts to operationalize the Blueprint framework across kernel optimization, RTL/PPA, verification,