Senior GPU Software Engineer

Senior GPU Performance Software Engineer About the Role

The Software and AI (SAI) organization is seeking a highly skilled software engineer to contribute to the development and low-level optimization of oneDNN, a complex, cross-platform, open-source performance library that serves as the foundation for deep learning applications.

Please Note:

This is a low-level software engineering and hardware-acceleration role. It does not involve building, training, or tuning machine learning models. Instead, you will focus on developing highly optimized math primitives, parallel algorithms, and GPU kernels that power industry-leading AI frameworks (such as OpenVINO, Tensor Flow, PyTorch, and ONNX Runtime).

• Develop high-performance GEMM, convolution, and attention kernels for AI workloads
• Design scalable JIT and codegen infrastructure for GPU kernel generation

• Implement fusion and memory-traffic optimizations to maximize hardware utilization
• Optimize mixed-precision and quantized execution paths (e.g., BF16, FP16, INT8, FP8, FP4, etc.)

• Build analytical and empirical performance models for kernel dispatch and tuning
• Profile and eliminate performance bottlenecks across oneDNN GPU primitives and runtime paths

• Co-design GPU primitives and kernel architectures for next-generation GPUs
• Partner with hardware and compiler teams to shape future accelerator capabilities and software stacks

• Improve validation, benchmarking, and CI infrastructure for performance-critical GPU workloads

• Work on a global, high-impact open-source library that scales AI performance across millions of devices worldwide

• Get early access to and influence the software stack for the roadmap of next-generation discrete GPUs

• Work alongside industry-leading experts in GPU compilers, hardware architecture, and performance libraries

• Enjoy a competitive package including stock programs, quarterly bonuses, robust healthcare, and highly flexible hybrid/remote working options

• A strong ownership mindset - you take initiative on complex, ambiguous technical problems and drive them to resolution
• A collaborative approach - you work effectively across hardware, compiler, and framework teams to align on shared technical goals
• A performance-driven curiosity - you are motivated by squeezing every cycle out of hardware and continuously seek deeper understanding of low-level systems

• Education: BSc, MSc, or PhD in Computer Science, Computer Engineering, Mathematics, Physics, or a highly technical related field
• Core Language: 5+ years of professional software development experience with expert-level modern C++
• Performance Optimizations: 2+ years of hands-on experience in programming and kernel optimization on GPUs (via SYCL/DPC++, OpenCL, CUDA, or HIP), or at least 5+ years of similar low-level performance optimization experience on CPUs
• Hardware Architecture: Strong foundations in computer architecture, cache hierarchies, memory subsystems, and parallel programming paradigms (e.g., multi-threading, SIMD/vectorization)

• Math Libraries:
  Experience developing high-performance math libraries (e.g., GEMM, convolution, reduction, or FFT kernels)
• Low-Level Tuning:
  Hands-on experience with GPU assembly-level tuning or compiler optimization
• Parallel APIs:
  Familiarity with parallel programming APIs such as OpenMP or oneTBB
• AI Workload Context:
  Basic understanding of deep learning primitives (e.g., forward/backward passes) to understand how library code is utilized by upstream frameworks

Experienced Hire

Shift 1 (United States of America)

US, Oregon, Hillsboro

US, California, Santa Clara

All qualified applicants will receive consideration for employment without regard to race, color, religion, religious creed, sex, national origin, ancestry, age, physical or mental disability, medical condition, genetic information, military and veteran status, marital status, pregnancy, gender, gender expression, gender identity, sexual orientation, or any other characteristic protected by local law, regulation, or ordinance.

N/A

We offer a total compensation package that ranks among the best in the industry. It consists of competitive pay, stock bonuses, and benefit programs which include health, retirement, and vacation. Find out more about the benefits of working at .

Annual Salary Range for jobs which could be performed in the US: $- USDThe range displayed on this job posting reflects the minimum and maximum target compensation for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant…