Lead Engineer, AI Infrastructure

Responsibilities

• Develop reliable, performant, and maintainable software for training Grok models in large-scale compute clusters.
• Provide technical direction for future training architectures and hardware designs.
• Collaborate with model, kernel, network, hardware design, platform, and site operations engineers.
• Develop prototypes to validate design concepts and quantify technical constraints.
• Lead team planning, stakeholder coordination, milestone development, and execution.

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 1 year of C/C++ experience at SpaceX.
• Experience developing, debugging, and deploying software used in real-world applications or projects.
• Knowledge of NVL72 GB300 racks, computer architecture, networking, mathematics, and physics.
• Strong skills in modeling, debugging, performance optimization, and unit testing.
• Creative problem-solving, analytical skills, and strong engineering fundamentals.

Benefits

• Equity and comprehensive medical, vision, and dental coverage.
• Access to a 401(k) retirement plan, short- and long-term disability insurance, life insurance, discounts, and other perks.