Lead AI Infrastructure Engineer - Scale Grok on Massive GPUs

Digitailor; Palo Alto (CA); On-site; USD 220,000 - 357,000; Full time

Benefits offered by this job

Equity and stock options

Comprehensive medical, vision, and 401

Disability & life insurance

Job summary

SpaceXSI is seeking a Lead Engineer for AI Infrastructure in Palo Alto to architect and maintain software for training Grok models on massive GPU clusters. You will lead a team, set technical direction, and ensure reliability, throughput, and security across the stack.

The role requires a strong background in C/C++, systems programming, and performance optimization, with collaboration across model, kernel, network, and hardware teams to push the boundaries of intelligence infrastructure.

Qualifications

• Bachelor’s degree in computer science, engineering, math, or technical discipline.
• Hands-on systems programming experience in C or C++.
• 3+ years of software engineering experience OR 1+ years of C/C++ experience at SpaceX.
• Solid fundamental knowledge of computer architecture, networking, math, and physics.
• Strong skills in modelling, debugging, performance optimization, and unit testing.

Responsibilities

• Develop reliable, performant, and maintainable software responsible for training Grok models in our world-class compute clusters.
• Provide technical direction for future training architectures and hardware designs.
• Work closely with other model engineers, kernel engineers, network engineers, hardware design engineers, platform engineers, and site ops engineers.
• Develop prototypes to prove out key design concepts and quantify technical constraints.
• Lead planning efforts for the team, working with key stakeholders in other teams and management to develop milestones and execute to them.

Skills

C/C++ programming

Systems programming

Performance optimization

Unit testing

Debugging

Modeling

Education

Bachelor’s degree in computer science or engineering

Job description

SpaceXSI is seeking a Lead Engineer for AI Infrastructure in Palo Alto to architect and maintain software for training Grok models on massive GPU clusters. You will lead a team, set technical direction, and ensure reliability, throughput, and security across the stack.

The role requires a strong background in C/C++, systems programming, and performance optimization, with collaboration across model, kernel, network, and hardware teams to push the boundaries of intelligence infrastructure.