AI Kernel Engineer

10 months ago

Burlingame, CA, USASenior

Base Salary

$110k - $270k/yr

Responsibilities

Develop AI and LLM kernels/operators on the Quadric platform for efficient inference.
Optimize kernel performance across hardware configurations and workloads.
Profile and analyze compute, data, and parallelism performance and identify micro-architecture and software bottlenecks.
Optimize C/C++ kernel code and maximize hardware utilization.
Collaborate across the AI inference stack to support team and business priorities.
Improve the Quadric toolchain, compiler, and runtime.
Provide technical support and documentation to customers and the developer community.

Bachelor’s or master’s degree in Computer Science and/or Electrical Engineering.
At least 5 years of experience in AI kernel development and optimization.
Experience with model and kernel inference performance profiling.
Experience with at least one of CUDA, DSP, NEON, or Triton-lang.
Proficiency in C/C++ and Python.
Assembly language experience is a plus.
Strong problem-solving, debugging, and communication capabilities.

Competitive salary and meaningful equity.
Medical, dental, and vision plan options starting on day one.
401(k) retirement plan.
Flexible unlimited paid time off with no accrual.
Company-provided lunches and a stocked kitchen when working in the office.
Support for commuting, including monthly parking or Caltrain passes.
Hybrid schedule with at least two in-office days per week expected in the Burlingame office, subject to team and business needs.
Periodic onsite meetings and offsite events.
Downtown Burlingame office near Caltrain, shops, cafes, and local amenities.

51-200 employees