Staff Compiler Engineer for AI/ML Hardware Backends

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

Benefits offered by this job

4+ weeks paid time off

Medical/Dental/Vision benefits

Flexible work arrangements

Onsite gym and café

Mental health support

Job summary

Samsung Semiconductor in San Jose is seeking an expert to join the AGI Computing Lab, focusing on AI/ML workloads. The ideal candidate will work on adapting compiler techniques for hardware, designing kernel DSLs, and contributing to open-source projects.

The role requires a Bachelor's with 10+ years or a higher degree with relevant experience, especially in Triton or MLIR. The position offers a competitive salary range and extensive benefits including 4+ weeks of PTO and flexible work arrangements.

Qualifications

• 10+ years of industry experience with a Bachelor's, 8+ with a Master's, or 5+ with a PhD.
• 3-5+ years in machine learning libraries like Triton or Helion.
• Experience designing kernel DSLs or making language-level changes.

Responsibilities

• Adapt torch.compile to hardware for efficient compilation.
• Build or extend kernel DSLs for hardware compatibility.
• Design placement and scheduling passes for memory management.
• Implement parallelism-aware lowering in the compiler.

Skills

Experience with Triton

Experience with MLIR

Experience with kernel autotuning

HPC background

Open-source contributions

Education

Bachelor's degree with 10+ years of experience

Master's degree with 8+ years of experience

PhD with 5+ years of experience

Tools

PyTorch

Triton

Helion

Job description

Samsung Semiconductor in San Jose is seeking an expert to join the AGI Computing Lab, focusing on AI/ML workloads. The ideal candidate will work on adapting compiler techniques for hardware, designing kernel DSLs, and contributing to open-source projects.

The role requires a Bachelor's with 10+ years or a higher degree with relevant experience, especially in Triton or MLIR. The position offers a competitive salary range and extensive benefits including 4+ weeks of PTO and flexible work arrangements.