X

xAI

Generative AI chatbot leveraging real-time data

Software Engineer - Kernels/CUDA, C++

Systems Software EngineeringConfirmed live in the last 24 hours
Full-timeIn-person

About the Job

Requirements

  • At least 1 year of experience with low-level system programming in C or C++.
  • Deep low-level systems programming experience in C, C++, PTX, or SASS.
  • Strong experience with large-scale GPU clusters or distributed compute infrastructure at production scale.
  • Hands-on experience with GPU kernel optimization using CUTLASS, custom kernels, or Nsight profiling.
  • A track record of building or running high-performance infrastructure for AI training or inference workloads.
  • Ability to reason from first principles and optimize both memory-bound and compute-bound scenarios.

Responsibilities

  • Design, build, and optimize massive GPU clusters for extreme-scale training and inference workloads.
  • Develop and tune low-level CUDA kernels, including GeMM and Attention, using CUTLASS, Tensor Cores, and Nsight for maximum performance.
  • Profile, debug, and eliminate bottlenecks across GPU memory hierarchy, networking fabric, filesystems, and multi-GPU operation.
  • Collaborate closely with AI research teams to deliver production-grade performance and scalability.

About the Company

xAI

Generative AI chatbot leveraging real-time data

xAI builds Grok, a generative AI chatbot with a Hitchhiker's Guide-inspired persona that uses real-time data from X to produce current, culturally aware responses. Grok is accessible to premium subscribers on X, via a standalone website, mobile apps, and through an API for developers. The company also develops large-scale AI infrastructure, such as the Colossus supercomputer, to train and run its models. Its goal is to pursue truth-seeking AGI research and grow a capital-intensive platform that could support a major public offering in the future.

Company size
1,001-5,000
Company stage
Series E
Total funding
$42.1B
Headquarters
Palo Alto, CA
Founded
2023

Recently Posted Jobs

Explore more opportunities at xAI

See more jobs