← Back to job listings
XA
Software Engineer (Kernels/CUDA (C++))
xAI · Seattle, United States
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
JobSpring