Skip to content
← Back to job listings

Software Engineer (Kernels/CUDA (C++))

xAI · Seattle, United States

External listingfull-time4 days ago

About The Role

Join our Compute Infrastructure team as a Software Engineer, where you will play a crucial role in building one of the world's largest AI supercomputers. You will work across the full stack, from low-level GPU kernel optimizations to massive-scale orchestration and virtualization. Your contributions will directly accelerate AI progress and improve training speed.

  • 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.
  • Work on Linux kernel internals, scheduling, memory management, and resource isolation at cluster scale.
  • Ability to reason from first principles and optimize for both memory-bound and compute-bound scenarios
  • Track record of building or running high-performance infrastructure for AI workloads (training or inference platforms)
  • Strong experience with large-scale GPU clusters or distributed compute infrastructure at production scale
  • Experience building and operating high performance exabyte scale storage systems
  • Experience with Linux kernel internals, scheduling, virtualization, or large-scale orchestration
  • Hands-on work with GPU kernel optimization (CUTLASS, custom kernels, Nsight profiling)
  • Deep low-level systems programming (C/C++ or Rust)

This is an external listing. JobSpring does not represent or verify the employer. Report this listing