Staff Compiler Engineer, PyTorch — Hybrid/Onsite IR & DSL

Samsung Semiconductor; San Jose (CA); On-site; USD 163,000 - 253,000; Full time

Benefits offered by this job

4+ weeks paid time off

Diverse benefits package for wellbeing

Support for family care and emotional wellness

Job summary

Samsung Semiconductor is seeking an experienced individual for an impactful role in San Jose, CA focused on developing innovative kernel DSLs and enhancing compiler performance for specific hardware. Candidates must possess significant industry experience, preferably with Triton or similar technologies.

This position offers a comprehensive benefits package, and an excellent salary range of $163,000—$253,000 depending on qualifications, along with a flexible work environment that supports overall well-being.

Qualifications

• Bachelor’s with 10+ years, Master’s with 8+ years, or PhD's with 5+ years of industry experience.
• 3-5+ years of industry experience in Triton, Helion, MLIR or similar.
• Experience designing a kernel DSL or making significant changes to an existing one.
• Experience building PyTorch backends for non-CUDA accelerators.
• Background in HPC, distributed systems, or NUMA-aware programming.

Responsibilities

• Adapting torch.compile to backend-specific requirements.
• Building or extending kernel DSLs for hardware.
• Designing placement and scheduling passes based on memory models.
• Implementing parallelism-aware lowering in the compiler IR.
• Contributing upstream to open-source projects.

Skills

Kernel DSL design

MLIR experience

PyTorch backend development

Performance modeling

Open-source contributions

Education

Bachelor’s degree with 10+ years, Master’s with 8+ years, or PhD with 5+ years of industry experience

Job description

Samsung Semiconductor is seeking an experienced individual for an impactful role in San Jose, CA focused on developing innovative kernel DSLs and enhancing compiler performance for specific hardware. Candidates must possess significant industry experience, preferably with Triton or similar technologies.

This position offers a comprehensive benefits package, and an excellent salary range of $163,000—$253,000 depending on qualifications, along with a flexible work environment that supports overall well-being.