Senior Software Engineer Kernels CUDA C++
SpaceXAI builds frontier AI models and products including the Grok conversational assistant and a multimodal developer API.
Funding history
About SpaceXAI
SpaceXAI is a US-based AI company focused on accelerating scientific discovery and understanding the universe through reasoning, voice, image, video, and generative AI systems. Formerly xAI, it was acquired by SpaceX effective February 2, 2026, while continuing to operate the x.ai site and services.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will design, build, and optimize GPU clusters for large-scale training and inference. You will develop CUDA kernels, profile and eliminate bottlenecks across GPU memory, networking, filesystems, and multi-GPU operations, and deliver scalable production performance with AI research teams.
Requirements
- 5 or more years of low-level systems programming experience in C/C++
- Low-level systems programming
- PTX
- SASS
- Large-scale GPU clusters
- Distributed compute infrastructure
- GPU kernel optimization
- CUTLASS
- Nsight profiling
- AI training infrastructure
- AI inference infrastructure
Responsibilities
- Design, build, and optimize massive GPU clusters for extreme-scale training and inference
- Develop and tune low-level CUDA kernels using CUTLASS, Tensor Cores, and Nsight
- Profile, debug, and eliminate bottlenecks across GPU memory, networking, filesystems, and multi-GPU operations
- Collaborate with AI research teams to deliver production-grade performance and scalability
Benefits
- Equity
- Comprehensive medical coverage
- Vision coverage
- Dental coverage
- 401(k) retirement plan
- Short-term disability insurance
- Long-term disability insurance
- Life insurance
- Discounts and perks