← All roles

Software Engineer - Kernels/CUDA (C++)

xAI· Infrastructure
Apply on company site
Hybrid (Palo Alto, CA; Seattle, WA)$180k - $440kFull-timePosted 7mo ago

About the role

Role within SpaceXSI building a world-scale AI supercomputer, owning the raw GPU system and platform layer, spanning low-level GPU kernel work to large-scale orchestration and virtualization to accelerate Grok’s training and AI progress.

Responsibilities

  • Design, build, and optimize massive GPU clusters for extreme-scale training and inference workloads
  • Develop and tune low-level CUDA kernels (GeMM, Attention, etc.), using CUTLASS, Tensor Cores, and Nsight for maximum performance
  • Profile, debug, and eliminate bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operation
  • Collaborate closely with AI research teams to deliver production-grade performance and scalability

Qualifications

  • Deep low-level systems programming (C/C++/PTX/SASS)
  • Strong experience with large-scale GPU clusters or distributed compute infrastructure at production scale
  • Hands-on work with GPU kernel optimization (CUTLASS, custom kernels, Nsight profiling)
  • Track record of building or running high-performance infrastructure for AI workloads (training or inference platforms)
  • Ability to reason from first principles and optimize for both memory-bound and compute-bound scenarios

Skills

Required:C/C++PTX/SASSCUDA kernel optimizationGPU cluster / distributed compute

Benefits

  • equity
  • comprehensive medical, vision, and dental coverage
  • 401(k) retirement plan
  • short & long-term disability insurance
  • life insurance
  • various other discounts and perks
Last updated Oct 14, 2026. Interested? Apply on the company site.

About xAI

liquid-ai-recruiting-target
Company profileWebsite

More open roles at xAI