Staff Software Engineer - GenAI Performance and Kernel

Tech Stack

Responsibilities

• Lead the design, implementation, benchmarking, and maintenance of core compute kernels optimized for various hardware backends (GPU, accelerators)
• Drive the performance roadmap for kernel-level improvements including vectorization, tensorization, tiling, fusion, mixed precision, sparsity, quantization, memory reuse, scheduling, and auto-tuning
• Integrate kernel optimizations with higher-level ML systems and build/maintain profiling, instrumentation, and verification tooling
• Lead performance investigations and root-cause analysis on inference bottlenecks
• Mentor and guide other engineers on lower-level performance, provide code reviews, and help set best practices

Soft Skills

Benefits

• Equity

Culture

Requirements