Staff Engineer, Compiler

The role

You will lead the architecture, design, and implementation of ML compilers for high-performance deep learning applications. You will work on staged lowering infrastructure and novel optimization algorithms for AI accelerator systems.

What you will do

• Lead the architecture, design, and implementation of ML compilers for high-performance deep learning applications
• Design and evolve a staged lowering infrastructure
• Invent and implement novel optimization algorithms
• Act as the technical bridge between compiler and hardware teams
• Influence and align cross-functional teams
• Raise the engineering bar through design and code reviews, mentoring, and setting technical standards
• Troubleshoot and debug complex compiler and performance issues
• Shape compiler documentation, user guides, and best practices

Required qualifications

• Bachelor’s with 10+ years of industry experience, or Master’s with 8+ years, or PhD's with 5+ years of industry experience
• Strong background in compiler design and optimization techniques
• Experiences in developing and optimizing software for high-performance computing systems
• Familiarity with PyTorch, Tensorflow, or JAX
• Familiarity with hardware architectures such as CPUs, GPUs, TPUs, and NPUs
• Strong analytical and problem-solving skills
• Excellent communication and interpersonal skills
• Ability to work independently and as part of a team
• Inclusive, adapting style to the situation and diverse global norms
• Avid learner
• Collaborative
• Innovative and creative

Nice to have

• Experiences in LLVM / MLIR

Benefits

• Incentive opportunities
• Medical/Dental/Vision
• 401k
• Charitable giving match
• 4+ weeks of paid time off a year, plus holidays and sick leave
• Stipend for fertility care or adoption
• Caregiver support
• Medical travel support
• On-demand apps and free confidential therapy sessions
• Onsite Café and gym
• Virtual classes
• Flexible environment

Before you apply

• Each candidate is limited to 10 applications across all open jobs within a 6-month period
• AI tools should not be used to generate or assist with submitted content or live interview responses
• By applying, you agree not to disclose confidential or proprietary information belonging to a current or former employer

About the company

Samsung Semiconductor is a global leader whose technology solutions power tools including smartphones, electric vehicles, hyperscale data centers, and IoT devices.