Senior Kernel & Compiler Performance Engineer (GPU/AI)

RadixArk

Palo Alto (CA)

On-site

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Competitive compensation
Comprehensive benefits
Flexible work arrangements

Job summary

A cutting-edge AI company in California is looking for a Member of Technical Staff for Kernel/Compiler/Communication. This critical role requires strong expertise in CUDA and GPU optimization, along with 5+ years of experience in performance engineering. The ideal candidate will design high-performance kernels and optimize systems for large GPU clusters, contributing to the next generation of AI solutions. The company offers competitive compensation, equity, and a collaborative work environment.

Qualifications

  • 5+ years of experience in systems, compiler, or performance engineering.
  • Strong expertise in CUDA or accelerator programming.
  • Deep understanding of GPU architecture and memory hierarchy.
  • Experience writing or optimizing high-performance kernels.
  • Strong background in compilers, runtimes, or code generation.

Responsibilities

  • Design and implement high-performance kernels for AI workloads.
  • Optimize compiler and runtime stacks for ML systems.
  • Improve communication efficiency across large GPU clusters.
  • Reduce latency and increase throughput for distributed workloads.
  • Profile and eliminate system bottlenecks across the stack.

Skills

CUDA or accelerator programming
GPU architecture and memory hierarchy
C++
Python
Debugging and profiling skills at system level

Job description

A cutting-edge AI company in California is looking for a Member of Technical Staff for Kernel/Compiler/Communication. This critical role requires strong expertise in CUDA and GPU optimization, along with 5+ years of experience in performance engineering. The ideal candidate will design high-performance kernels and optimize systems for large GPU clusters, contributing to the next generation of AI solutions. The company offers competitive compensation, equity, and a collaborative work environment.
Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Staff Engineer: GPU Kernels & AI Performance
Staff Engineer: GPU Kernels & AI Performance

Gimlet Labs • San Francisco (CA)

On-site
USD 120,000 - 160,000
Senior AI Kernel Engineer — GPU Inference & Kernel Optimization
Senior AI Kernel Engineer — GPU Inference & Kernel Optimization

Modular • United States

Hybrid
USD 198,000 - 286,000
Staff GPU Kernel & Performance Engineer
Staff GPU Kernel & Performance Engineer

Gimlet Labs, Inc. • San Francisco (CA)

On-site
USD 150,000 - 350,000
Remote CUDA Kernel Engineer - AI GPU Performance
Remote CUDA Kernel Engineer - AI GPU Performance

Pragmatike • New Jersey

On-site
USD 110,000 - 150,000
GPU Kernel Engineer for AI Inference & Performance
GPU Kernel Engineer for AI Inference & Performance

FriendliAI • San Francisco (CA)

On-site
USD 120,000 - 150,000
Flexible working hours
Daily lunch and dinner
Health check-up support
+3
GPU Kernel Engineer for High-Performance AI Inference
GPU Kernel Engineer for High-Performance AI Inference

Baseten • San Francisco (CA)

On-site
USD 180,000 - 360,000
Competitive compensation, including equity
100% coverage of medical, dental, and vision insurance
Flexible PTO policy
+3
GPU Kernel Engineer: Build Fast AI Inference at Scale
GPU Kernel Engineer: Build Fast AI Inference at Scale

Baseten • San Francisco (CA)

On-site
USD 120,000 - 160,000
Competitive compensation
100% medical coverage
Generous PTO policy
+2
Staff GenAI Kernel & Performance Engineer
Staff GenAI Kernel & Performance Engineer

Databricks • San Francisco (CA)

On-site
USD 190,000 - 233,000
AI Kernel Engineer: High-Performance Compute
AI Kernel Engineer: High-Performance Compute

Majestic Labs ai • Los Altos (CA)

On-site
USD 70,000 - 90,000
Kernel Performance Engineer for AI Tooling
Kernel Performance Engineer for AI Tooling

OpenAI • Los Angeles (CA)

On-site
USD 100,000 - 130,000