Apply on Employer Site

AMD · 1 day ago

Performance Engineer (ML)

San Jose, CA

Full-time

Onsite

Senior Level

$192K/yr - $288K/yr

5+ years exp

AMD is a company focused on building products that enhance computing experiences across various domains. They are seeking an experienced ML Compiler Engineer to optimize performance for machine learning workloads on their datacenter GPU infrastructure, working at the intersection of compiler technology and machine learning.

AI InfrastructureArtificial Intelligence (AI)Cloud ComputingComputerEmbedded SystemsGPUHardwareSemiconductor

Growth Opportunities

No H1B

Responsibilities

Identify and analyze performance bottlenecks in ML workloads at the compiler level, using profiling tools and performance analysis methodologies

Design and implement optimization passes in the LLVM backend to improve code generation quality for datacenter GPU architectures

Develop and enhance instruction selection, scheduling, and register allocation strategies specific to ML computation patterns

Collaborate with hardware architecture teams to ensure compiler optimizations align with and fully exploit GPU microarchitectural features

Contribute bug fixes, optimizations, and new features to upstream LLVM projects, following open source development practices

Maintain and improve internal LLVM-based compiler toolchains, ensuring compatibility with upstream changes while adding proprietary enhancements

Create comprehensive benchmarks and test suites to validate compiler optimizations and prevent performance regressions

Document compiler optimizations, architectural decisions, and best practices for both internal and external audiences

Work with ML framework teams to understand and optimize compilation of real-world models and workloads

Participate in code reviews, design discussions, and technical presentations to share knowledge across the organization

Qualification

LLVM backend developmentPerformance optimizationCompiler technologyComputer architectureOpen-source contributionAnalytical skillsEffective communicationProblem-solving skills

Required

Experienced ML Compiler Engineer

Deep expertise in LLVM backend development

Proven track record of optimizing code generation for specialized hardware architectures

Strong analytical skills to diagnose performance issues at the compiler level

Ability to devise creative solutions for performance optimization

Active participation in the open source community

Understanding of collaborative development and code review processes

Ability to communicate technical concepts clearly

Effective collaboration with cross-functional teams including hardware architects and ML researchers

Driven by the impact of optimizations on real-world AI workloads running at scale

Identify and analyze performance bottlenecks in ML workloads at the compiler level using profiling tools and performance analysis methodologies