Lead AI Infra Architect for Scalable GPU Training

SpaceXAI

Palo Alto, Seattle (CA, WA)

On-site

USD 220,000 - 357,000

Full time

6 days ago

Be an early applicant

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Equity

Comprehensive medical, vision, and 401

Disability insurance

Life insurance

Retirement plan

Employee discounts

Job summary

SpaceXSI is seeking a Lead Engineer for AI infrastructure in Palo Alto to architect, implement, and maintain the software framework that trains Grok models on massive GPU clusters. You will lead a small team, drive reliability, fault tolerance, throughput, security, and developer experience while shipping scalable infrastructure.

The role demands strong C/C++ systems programming skills, hands-on leadership, and collaboration with model, kernel, and hardware engineers to realize the Grok

Qualifications

Bachelor's degree in a technical discipline is required.
Hands-on systems programming in C or C++.
Experience with large-scale GPU clusters is a plus.

Responsibilities

Architect and maintain software infrastructure for training Grok models on GPU clusters.
Provide technical direction for future training architectures and hardware designs.
Collaborate with model, kernel, network, and hardware engineers, plus site ops.
Develop prototypes to prove design concepts and quantify constraints.
Lead team planning with stakeholders to set milestones and execute.

Skills

C/C++ programming

Systems programming

Performance optimization

Debugging

Distributed compute knowledge

Education

Bachelor's degree in CS/Engineering/Math

Job description

The role demands strong C/C++ systems programming skills, hands-on leadership, and collaboration with model, kernel, and hardware engineers to realize the Grok

Get your free, confidential resume review.

or drag and drop your file here.