ML Accelerator Compiler Engineer

Black Sesame; San Jose, Northern (CA, KY); Hybrid; USD 150,000 - 190,000; Full time

Job summary

Black Sesame is seeking an ML Accelerator Compiler Developer in San Jose to design and optimize compiler toolchains for ML accelerators, focusing on high performance and portability. You will implement ML-specific optimizations and collaborate with hardware teams to tailor compiler passes for target architectures.

The role involves working with ML frameworks like PyTorch and ONNX, driving operator fusion, quantization-aware compilation, and scheduling, while profiling and debugging across

Qualifications

• Strong proficiency in compiler development with LLVM/MLIR/TVM experience.
• Expertise in ML model execution, optimization and deployment.
• Strong programming in C++, Python, and assembly-level optimizations.
• Knowledge of parallel computing, vectorization and memory hierarchies.
• Familiarity with TensorFlow, PyTorch, ONNX.
• Strong analytical skills for performance profiling and debugging.
• Experience in graph optimizations, quantization, and code generation.

Responsibilities

• Design and optimize compilers for ML accelerators.
• Develop and optimize compiler toolchains, IR transformations, and backend code generation.
• Implement ML-specific optimizations such as operator fusion, memory layout transformations, quantization-aware compilation, and scheduling.
• Collaborate with hardware architects to co-design compiler optimizations aligned with accelerator capabilities.
• Work on ML frameworks (PyTorch, ONNX) to integrate compiler passes for efficient execution on target hardware.
• Improve performance through domain-specific optimizations, autotuning, and parallelization techniques.
• Debug and analyze performance bottlenecks across software and hardware stacks.
• Develop automated testing, benchmarking, and profiling tools for validating compiler optimizations.

Skills

Compiler development

ML model execution

C++

Python

Assembly optimizations

Parallel computing

Vectorization

DL frameworks

Profiling & debugging

Graph optimizations

Quantization

Code generation

Job description

Black Sesame is seeking an ML Accelerator Compiler Developer in San Jose to design and optimize compiler toolchains for ML accelerators, focusing on high performance and portability. You will implement ML-specific optimizations and collaborate with hardware teams to tailor compiler passes for target architectures.

The role involves working with ML frameworks like PyTorch and ONNX, driving operator fusion, quantization-aware compilation, and scheduling, while profiling and debugging across