Principal Software Development Engineer - AI/ML

Advanced Micro Devices, Inc.
San Jose, CA, United States
2 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours

Tech stack

Artificial Intelligence C++ (Programming Language) Compilers Profiling Nvidia CUDA Software Debugging Linux Object-Oriented Software Development Performance Tuning Tensorflow Software Engineering Graphics Processing Unit (GPU)
+3 more
Gpu Programming Git ONNX (Open Neural Network Exchange) Format

Job description

Experteer Overview In this role, you will optimize AI workloads on AMD GPUs through advanced compiler work and performance engineering. You’ll bridge architecture, compilers, and ML frameworks to enhance ROCm and ONNX Runtime execution. You will collaborate across architecture, kernel, runtime, and framework teams to deliver scalable AI solutions. This position offers the chance to shape AI performance across current and future GPU platforms. Join a mission-driven team pushing the boundaries of AI on AMD hardware. Compensation / Benefits * Collaborate with architecture specialists to shape future products * Design, develop, and optimize compiler transformations and operator lowerings within MLIR, LLVM, and ROCm * Create scalable solutions for portability and performance across AMD GPUs * Work with compiler and runtime teams on advanced AI workload optimizations * Implement, validate, and maintain ONNX operators and microbenchmark suites * Lead performance optimization efforts for AI training, fine-tuning, RL, and inference on AMD GPUs * Design and implement graph-, kernel-, and runtime-level optimizations yielding measurable gains * Analyze workload characteristics to identify compute, memory, and communication improvements * Drive performance characterization and benchmarking pre-silicon and post-silicon Tasks * Strong proficiency in C++ and object-oriented software development * Hands-on GPU programming experience (HIP, CUDA, Triton, or equivalent) * Experience with MLIR, LLVM, or related compiler infrastructures * Solid understanding of ML frameworks, model execution pipelines, and AI workload optimization * Experience profiling and optimizing performance-critical software * Proficiency with Linux, Git, debugging tools, and performance profilers * Strong analytical, communication, and problem-solving skills Key requirements *

Requirements

training, fine-tuning, RL, and inference on AMD GPUs * Design and implement graph-, kernel-, and runtime-level optimizations yielding measurable gains * Analyze workload characteristics to identify compute, memory, and communication improvements * Drive performance characterization and benchmarking pre-silicon and post-silicon Tasks * Strong proficiency in C++ and object-oriented software development * Hands-on GPU programming experience (HIP, CUDA, Triton, or equivalent) * Experience with MLIR, LLVM, or related compiler infrastructures * Solid understanding of ML frameworks, model execution pipelines, and AI workload optimization * Experience profiling and optimizing performance-critical software * Proficiency with Linux, Git, debugging tools, and performance profilers * Strong analytical, communication, and problem-solving skills Key requirements *

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on us.experteer.com

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

2:33 min

Architecting CUDA and the AI software stack

Michael Kagan Michael Kagan +1 · World Congress 2026 Europe

6:21 min

Investigating push inefficiencies with upstream Git experts

Jonathan Creamer · Coffee With Developers

52 sec

Running persistent Linux environments directly on Windows

Ben Breard Ben Breard · World Congress 2025

1:51 min

Evolution of custom compilers and virtual machines

Florian Rappl · LIVE

4:20 min

Utilizing AI and hardware acceleration for application code optimization

Stephan Gillich Stephan Gillich · World Congress 2024

56 sec

Favorite git commands and the importance of patch commits

Eileen Uchitelle Eileen Uchitelle +1 · Coffee With Developers

Videos

See all

Related articles

See all