← All roles

Sr. AI/ML Platform Engineer

AMD· Engineering
Apply on company site
Santa Clara, CA$204kFull-timePosted 3mo ago

About the role

Senior AI/ML platform engineer responsible for building and operating the shared AI platform to support large-scale agent execution, distributed training/inference, experiment tracking, and reproducible workflows across GPU clusters, with collaboration across ML systems research, applied AI, and hardware teams.

Responsibilities

  • Build and operate the shared AI platform for agentic engineering workflows, including job submission, scheduling, orchestration, retries, logging, artifact storage, and experiment tracking.
  • Develop reliable infrastructure for distributed training, distributed inference, batch evaluation, and large-scale agent rollout across GPU clusters.
  • Build platform services for benchmark execution, correctness checking, profiling, regression tracking, and reproducible evaluation.
  • Maintain artifact systems for generated kernels, RTL edits, traces, logs, profiler outputs, benchmark results, simulator outputs, and formal verification artifacts.
  • Support scalable integrations with compilers, ROCm/HIP tooling, profilers, simulators, EDA tools, vLLM, SGLang, and internal engineering systems.
  • Improve GPU cluster utilization, scheduling efficiency, reliability, quota management, and workload isolation.
  • Build dashboards and observability systems for experiment status, resource usage, failure modes, benchmark trends, regression detection, and team productivity.
  • Partner with ML Systems Research Engineers to productionize research workflows for RL systems, inference systems, quantification systems, and evaluation pipelines.
  • Partner with Applied AI Engineers to make Blueprint harnesses reusable across kernel optimization, RTL optimization, verification, firmware, and CPU/GPU performance workflows.
  • Establish platform standards for reproducibility, data retention, run metadata, artifact lineage, access control, and operational reliability.

Qualifications

  • Strong programming skills in Python and one or more systems languages such as C++, Go, or Rust.

Skills

Required:PythonC++/Go/RustDistributed systems
Nice to have:GPU infrastructureKubernetes/Ray/SlurmExperiment tracking / artifact management

Benefits

  • Benefits described in AMD benefits at a glance.
Last updated Oct 22, 2026. Interested? Apply on the company site.
AMD logo

About AMD

together we advance_AI

AMD delivers leadership high-performance and adaptive computing solutions to advance data center AI, AI PCs, intelligent edge devices, gaming, & beyond.

AIData CenterHigh-Performance ComputingGamingEdge Devices
Company profileWebsite

More open roles at AMD