All Jobs
AMD

Software Development Engineer – AI Group

AMD|San Jose, United StatesOnsite
$198k - $198kUSDVerified
Apply Now

Job Description

Senior technical contributor in AMD’s AI group, responsible for end‑to‑end delivery of software solutions that optimize training and inference of ML models on AMD GPUs, collaborating across compiler, driver, and hardware teams.

Responsibilities

  • Analyze, implement, and optimize DirectML, TensorFlow, and PyTorch workloads on AMD GPUs
  • Coordinate with ISV, library, compiler, driver, and hardware teams to improve ML performance
  • Innovate new algorithms and kernel‑level optimizations exploiting AMD hardware strengths
  • Contribute to end‑to‑end solutions for training and inference workloads in data‑center and edge settings

Benefits

Base salary around $198,100/year plus benefitsExposure to cutting‑edge GPU and AI systems workHybrid culture signal (#LI-HYBRID) but role is primarily on‑site in San Jose

Category

Machine Learning Engineer

Posted

12/8/2025

Ready to Apply?

Applications go directly to AMD's career portal

Apply on AMD