Staff Engineer, Compiler

Samsung Semiconductor

Confirmed live today High trust

Quick summary

Work type
On-site
Location
San Jose, CA
Salary
$163,000–$253,000 / yr
Posted
147 days ago
Freshness
Confirmed live today
Nearby
99+ roles within 25 mi

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $194k
This role $208k
$128k $279k
below market most similar roles pay here above market

This role pays more than 50% of similar roles. Most pay $171,000–$217,906 — the blue band above. At the midpoint, this role pays about $208k versus about $194k for comparable roles.

Based on 240 similar postings.

Employer

About Samsung Semiconductor

Samsung Semiconductor is the global semiconductor business unit of Samsung Electronics, designing and manufacturing memory chips, logic semiconductors, and foundry solutions for a broad range of applications.

Samsung Semiconductor currently has 36 open roles on FindRole.

Listed pay typically runs $163,000–$253,000 across 36 roles with salary data.

Most-posted roles

View all roles at Samsung Semiconductor

At a glance

TL;DR · Staff Engineer, Compiler

Role Posting Log in to save

JOB TITLE: Staff Engineer, Compiler The Staff Engineer, Compiler joins the AGI Computing Lab to address complex system-level challenges for future AI/ML workloads. You will adapt torch.compile to the hardware backend by lowering Inductor's IR, defining fusion and specialization rules, and building or extending kernel DSLs. Key responsibilities include designing placement and scheduling passes for distributed memory models, implementing parallelism-aware lowering for tensor and sequence parallelism, and managing fusion, tiling, and memory planning for non-uniform memory hierarchies. You will utilize technologies including MLIR, Triton, Helion, XLA, TVM, Inductor, IREE, and CUTLASS. The role requires building PyTorch backends for non-CUDA accelerators and making upstream contributions to open-source projects. You will solve problems regarding computational and memory requirements while maximizing performance and minimizing energy consumption for demanding AI/ML applications.

What does a Engineer earn in California?

Median $199500 from 103 postings across 31 companies.

See salary data

Skills

PyTorch MLIR Triton XLA TVM Inductor IREE CUTLASS LLVM CUDA ROCm MPS TPU HPC Distributed Systems NUMA Kernel DSL Compiler IR torch.compile

What you'll do

  • Adapt torch.compile to the hardware backend by lowering Inductor's IR and defining fusion and specialization rules.
  • Build or extend kernel DSLs to lower high-level expressions to the hardware ISA and memory hierarchy.
  • Design placement and scheduling passes to manage tensor locations and overlap compute with data movement.
  • Implement parallelism-aware lowering for tensor, pipeline, expert, and sequence parallelism within the compiler IR.
  • Perform fusion, tiling, and memory planning for non-uniform memory hierarchies.
  • Contribute to upstream open-source projects including PyTorch, Triton, and Helion.

What we're looking for

  • Bachelor’s with 10+ years, Master’s with 8+ years, or PhD with 5+ years of industry experience.
  • 3-5+ years of industry experience in Triton, Helion, MLIR, XLA, TVM, Inductor, IREE, CUTLASS, or a proprietary equivalent.
  • Experience designing a kernel DSL or its IR from scratch, or making non-trivial language-level changes to an existing one.
  • Experience with MLIR, including writing dialects, passes, or backend integration.
  • Experience building PyTorch backends for non-CUDA accelerators such as XPU, ROCm, MPS, TPU, or custom hardware.
  • Experience with kernel autotuning, performance modeling, or cost-based compilation.
  • Background in HPC, distributed systems, or NUMA-aware programming.
  • Open-source contributions to PyTorch, Triton, Helion, LLVM/MLIR, or similar projects (preferred).

Related searches

More Staff Engineer jobs Staff Engineer jobs in San Jose All jobs at Samsung Semiconductor

More like this

Similar roles

Machine Learning Compiler Engineer

Qualcomm

Remote (New York, NY) +3 40 days ago $223,600–$335,400
MLIR LLVM PyTorch Triton TVM Generative AI Polyhedral Optimization High-Performance Computing GPU Programming Parallel Programming Vectorization Compiler Development Computer Architecture Quantized Neural Operators
8+ yrs exp Remote
Save

Senior Staff LLM Serving Engineer

Qualcomm

San Diego, CA +1 40 days ago $162,000–$243,000
LLM Serving PyTorch Python Triton-Inference Server vLLM SGLang CUDA Triton torch.compile torchDynamo Distributed Systems Deep Learning Model Optimization KV-Cache Management Inference Acceleration KServe Ollama LMCache MoonCake
4+ yrs exp
Save

PhD Large Language Model Engineer Co-op

Amd

San Jose, CA 22 days ago $91,520–$137,280
Large Language Models (LLMs) Machine Learning High-Performance Computing (HPC) Distributed Systems PyTorch TensorFlow JAX Python C++ CUDA MPI NCCL DeepSpeed Megatron-LM Triton XLA MLIR Quantization Model Parallelism FSDP
Save

Software Engineer, DL Kernel Libraries

Nvidia

Santa Clara, CA 16 days ago
CUDA C/C++ Python C++ PyTorch JAX TensorFlow vLLM SGLang cuDNN FlashInfer Triton MLIR Apache TVM ONNX XLA CUTLASS GPU Kernel Development Domain Specific Compilers MLC
Save

Staff Engineer, Software

Abbott

Sylmar, CA 81 days ago $114,000–$228,000
Java Object-Oriented Programming Git Jira Linux Generative AI Multi-threading Data Structures Performance Testing Source Code Management Embedded Systems UI/UX Design
8+ yrs exp
Save

Staff Engineer

GEICO

Remote (Seattle, WA) 161 days ago $115,000–$230,000
Docker Kubernetes Java .NET Go Python C++ HTML CSS JavaScript MySQL PostgreSQL BMC Helix REST SOAP CI/CD Infrastructure as Code Terraform Ansible Git CircleCI Grafana Prometheus AWS GCP Azure Lambda Kappa ML AI
6+ yrs exp Remote
Save