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

SpaceXAI builds frontier AI models and products including the Grok conversational assistant and a multimodal developer API.

Austin, United States
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.

View jobs by SpaceXAI

Skills

Candidate Availability

Required and preferred rules are kept separate and reflect the wording in the original posting.

About the Role

You will lead a team that architects, implements and maintains the software infrastructure used to train Grok models on very large GPU clusters. You will provide technical direction for future training architectures and hardware designs, build prototypes to prove design concepts, collaborate with model, kernel, network, hardware, platform and site ops engineers, and lead planning and milestones with stakeholders.

Requirements

  • Bachelor's degree in computer science, engineering, math, or technical discipline
  • Hands-on systems programming experience in C or C++
  • Preferred: 3+ years of software engineering experience or 1+ years of C/C++ experience at SpaceX
  • Preferred: knowledge of NVL72 GB300 racks, computer architecture, networking, math and physics
  • Preferred: skills in modelling, debugging, performance optimization and unit testing

Responsibilities

  • Develop reliable, performant and maintainable software for training Grok models
  • Provide technical direction for future training architectures and hardware designs
  • Work closely with model, kernel, network, hardware design, platform and site ops engineers
  • Develop prototypes to prove key design concepts and quantify technical constraints
  • Lead team planning and milestones with stakeholders and management

Benefits

  • Equity
  • Comprehensive medical, vision, and dental coverage
  • 401(k) retirement plan
  • Short & long-term disability insurance
  • Life insurance
  • Discounts and perks