Lead Engineer, AI Infrastructure (C++, C)
xAI, Palo Alto, CA +1
The role
The Grok training program is looking for a lead software engineer on the AI infrastructure team. You will lead the team that architects, implements, and maintains the software infrastructure responsible for training grok models on hundreds of thousands or millions of GPUs.
The success of Grok depends on the quality of the software that you and your team produce, including reliability, fault tolerance, throughput, security, and developer experience. You are the engineer that other engineers count on; you are technical, you tackle every problem with enthusiasm; you remove hurdles. You are a self-starter and a team player. We are seeking a talented lead software engineer to join us in making our vision a reality.
What you will do
- 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