A

AMD

Designs chips for computers and AI systems

Software Development Engineer - AI Workload Optimization

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeSeattle, WA, USANo salary listedIn-person

About the Job

Requirements

  • Deep software development experience and experience in a high-level technical leadership role, such as Fellow or equivalent.
  • Deep expertise in AI frameworks including PyTorch, JAX, vLLM, and SGLang, and the ROCm software stack.
  • Proven history of optimizing distributed inference and training at scale across multi-node, multi-GPU environments.
  • Mastery of performance profiling tools such as TorchProfiler, ROCm Profiler, and Nsight, and hardware-level performance modeling.
  • Strong understanding of modern model architectures, including Transformer, Attention, and KV Cache, and optimization techniques such as quantization, speculative decoding, and FlashAttention.
  • A Master's or Bachelor's degree in Computer Science, Electrical Engineering, or a related field, or equivalent experience.
  • Demonstrated research or applied experience in AI/ML, including deep learning, model training/inference optimization, large language models, or computer vision.

Responsibilities

  • Set the technical vision and roadmap for workload optimization across the AI software stack, ensuring AMD remains the platform of choice for top-tier AI customers.
  • Lead profiling, analysis, and tuning of large-scale models, including LLMs, diffusion, multimodal, and MoE models, to achieve strong out-of-the-box performance on AMD hardware.
  • Partner with top customers and hyperscalers to understand their workload requirements and deliver tailored architectural improvements and software optimizations.
  • Collaborate across hardware architecture, compiler, and framework teams to influence future silicon features based on evolving AI workload trends.
  • Drive development of advanced tools and frameworks for performance estimation, modeling, and automated reporting.
  • Represent AMD in industry forums and open-source communities, and mentor and inspire AMD's technical leaders and engineers.

About the Company

AMD

Designs chips for computers and AI systems

AMD designs processors, graphics chips and AI accelerators for computer makers, cloud operators and businesses that build electronic devices. Its Ryzen processors power personal computers, while EPYC processors handle work in data centers and Radeon chips render graphics. Instinct accelerators run AI models, paired with software that developers use to train models and put them into production. AMD sells chips and related computing systems through equipment makers, distributors and retail channels, and contracts with outside foundries to fabricate its semiconductor designs. It was founded in 1969 and has its headquarters in Santa Clara, California.

Company size
10,001+
Company stage
IPO
Headquarters
Santa Clara, CA
Founded
1969

Recently Posted Jobs

Explore more opportunities at AMD

See more jobs