Lead Engineer, AI Infrastructure (C++, C)

xAI•Seattle, WA

About The Position

SpaceXSI is building Super Intelligence to help humanity understand the universe, developing the full Super Intelligence stack from Colossus supercomputers and Grok models to applications including Grok Bot, Grok Build, Grok Imagine, Grok Voice, and X. The Lead Engineer, AI Infrastructure will lead the team responsible for architecting, implementing, and maintaining the software infrastructure used to train Grok models on hundreds of thousands or millions of GPUs. The role is accountable for the reliability, fault tolerance, throughput, security, and developer experience of the software produced by the team. The successful candidate provides technical leadership, removes obstacles, and works as a self-starting team player.

Requirements

Bachelor’s degree in computer science, engineering, mathematics, or another technical discipline.
Hands-on systems programming experience in C or C++.
At least 3 years of software engineering experience, or at least 1 year of C/C++ experience at SpaceX.
Strong fundamental knowledge of NVL72 GB300 racks, computer architecture, networking, mathematics, and physics.
Strong skills in modeling, debugging, performance optimization, and unit testing.
Creative problem-solving approach, exceptional analytical skills, and strong engineering fundamentals.

Nice To Haves

Experience developing, debugging, and deploying software used in real-world applications or projects.

Responsibilities

Develop reliable, performant, and maintainable software for training Grok models in compute clusters.
Provide technical direction for future training architectures and hardware designs.
Work closely with model, kernel, network, hardware design, platform, and site operations engineers.
Develop prototypes to validate key design concepts and quantify technical constraints.
Lead team planning with stakeholders in other teams and management, developing milestones and executing against them.