3 days ago
Base Salary
$220k - $357k/yr
Responsibilities
- Design, build, and optimize massive GPU clusters for large-scale training and inference workloads.
- 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.
Requirements
- At least 5 years of experience with low-level systems programming in C/C++.
- Deep low-level systems programming experience with C, C++, PTX, and SASS.
- Experience with large-scale GPU clusters or distributed compute infrastructure at production scale.
- Hands-on experience optimizing GPU kernels with CUTLASS, custom kernels, and Nsight profiling.
- Experience building or operating high-performance infrastructure for AI training or inference workloads.
- Ability to optimize memory-bound and compute-bound workloads from first principles.
Benefits
- Equity and comprehensive medical, vision, and dental coverage.
- Access to a 401(k) retirement plan, short- and long-term disability insurance, life insurance, discounts, and other perks.
About xAI
Understand the Universe. We are a team of AI technologists and business leaders on a mission to build AI systems that can help humanity understand the world better. https://x.ai/careers