Open Source ML Systems Engineer: GPU Kernels & Inference

Advanced Micro Devices; San Jose, Northern (CA, KY); Hybrid; USD 180,000 - 230,000; Full time

Job summary

Advanced Micro Devices is seeking an Open Source Software Engineer to build and optimize machine learning systems that integrate with AMD hardware. You will work at the intersection of open source AI software, GPU systems, and high-performance computing, bringing cutting-edge AI workloads to AMD platforms.

You will design, implement, test, and ship open source integrations with widely used AI systems, focusing on LLM inference, kernel optimization, and upstream contributions in collaborative

Qualifications

• Strong familiarity with LLM systems, inference workloads, compute, and optimization.
• Experience optimizing GPU kernels, particularly for AMD architectures.
• Ability to learn and adapt across kernel programming languages and models.

Responsibilities

• Develop and maintain open source integrations between ML/AI software and AMD hardware.
• Optimize kernels and critical execution paths for AMD GPUs.
• Profile workloads, identify performance bottlenecks, and implement optimizations.
• Develop kernel fusions and GPU execution pipelines.
• Adapt implementations across different kernel programming languages and environments.
• Collaborate with LLM inference systems and understand performance impacts of data layout and quantization.
• Contribute high-quality code upstream to open source projects.
• Write unit, integration, and performance tests.
• Build and maintain automated testing and validation infrastructure.
• Participate in code reviews with maintainers and contributors.

Skills

LLM workloads

GPU kernel optimization

Open source development

Performance profiling

Cross-layer optimization

Tools

Code review

Automated testing

CI

Upstream contribution

Job description

Advanced Micro Devices is seeking an Open Source Software Engineer to build and optimize machine learning systems that integrate with AMD hardware. You will work at the intersection of open source AI software, GPU systems, and high-performance computing, bringing cutting-edge AI workloads to AMD platforms.

You will design, implement, test, and ship open source integrations with widely used AI systems, focusing on LLM inference, kernel optimization, and upstream contributions in collaborative