Lead GPU Kernel Engineer for High-Performance ML

Coda Robotics; San Francisco (CA); On-site; USD 100,000 - 120,000; Full time

Job summary

Coda Robotics is looking for an experienced engineer to join their founding team, focusing on low-level compute kernels to enhance robotic foundation models. The ideal candidate will have substantial experience in systems programming (C/C++, assembly), expertise in GPU optimizations, and familiarity with ML framework internals. Responsibilities include leading engineering teams, integrating kernel optimizations, and pioneering high-velocity development culture. Compensation includes a salary range of $100,000 - $120,000 plus equity and benefits.

Qualifications

• Proven experience in low-level systems programming (C/C++, assembly) targeting CPU and GPU architectures.
• Expertise in developing and optimizing compute kernels (CUDA, ROCm, OpenCL, SIMD intrinsics).
• Deep understanding of performance profiling tools (nvprof, perf, Intel VTune).

Responsibilities

• Lead a team of kernel and system engineers focused on performance-critical code.
• Design, implement, and optimize custom compute kernels for CPU and GPU.
• Integrate kernel optimizations into distributed ML frameworks (e.g., PyTorch, TensorFlow).

Skills

Low-level systems programming

CUDA

C/C++

Memory optimization

GPU architectures

Performance profiling tools

Job description

Coda Robotics is looking for an experienced engineer to join their founding team, focusing on low-level compute kernels to enhance robotic foundation models. The ideal candidate will have substantial experience in systems programming (C/C++, assembly), expertise in GPU optimizations, and familiarity with ML framework internals. Responsibilities include leading engineering teams, integrating kernel optimizations, and pioneering high-velocity development culture. Compensation includes a salary range of $100,000 - $120,000 plus equity and benefits.