xAI
Software Engineer - Kernels/CUDA (C++)
Palo Alto, CA; Seattle, WA
Sponsorship not specified$180k-$440kDetected 1 day ago
C++SassFull-Stack DevelopmentLinuxResearchLeadershipCommunication
About the role
- You will work across the full stack - from low-level GPU kernel optimizations and Linux kernel internals to massive-scale orchestration and virtualization - to make training and inference at SpaceXAI as fast, reliable, and scalable as possible.
- This is a broad, high-impact role that combines hardcore supercompute and compute infrastructure work.
- Your contributions will directly accelerate Grok's training speed and overall 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
- Collaborate closely with AI research teams to deliver production-grade performance and scalability
- 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
- 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)
- This organization is for individuals who appreciate challenging themselves and thrive on curiosity.
- All employees are expected to be hands-on and to contribute directly to the company's mission.
- Work ethic and strong prioritization skills are important.
- All employees are expected to have strong communication skills.
- They should be able to concisely and accurately share knowledge with their teammates.
Compensation
- $180,000 - $440,000 USD
- Base salary is just one part of our total rewards package at SpaceXAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.
Company info
- We are building one of the world's largest AI supercomputers from the ground up.
- As part of the Compute Infrastructure team, you will own both the raw GPU supercomputer and the platform layer that runs on top of it.
Equal opportunity
- SpaceXAI is an equal opportunity employer.
- equal opportunity employer.
This listing is sourced directly from xAI's careers page and normalized into a canonical job model.