AI Engineer & Researcher - Multimodal

We are looking for a motivated engineer to join xAI, where our mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly driven, and focused on engineering excellence. We offer an environment for individuals who appreciate challenging themselves and thrive on curiosity. All engineers and researchers share the title 'Member of Technical Staff,' which fosters a flat organizational structure. Employees are expected to be hands-on and directly contribute to the company’s mission. Leadership opportunities are awarded to those who show initiative and consistently deliver excellence, while strong work ethic and prioritization skills are essential. Effective communication is a key requirement, as all team members must be able to concisely and accurately share knowledge.

Tech Stack:

• Python
• JAX and XLA
• Rust

Location:

• Based in the Bay Area (San Francisco and Palo Alto)
• Candidates should be located near the Bay Area or open to relocation

Focus:

• Training trillion parameter neural networks at scale, along with a variety of smaller specialized models
• Mastering all data modalities: text, audio, images, video, etc.
• Rapidly implementing the latest state-of-the-art methods from deep learning literature
• Innovating ideas that bring us closer to our goal of developing AI systems that can accurately understand the universe and generate new knowledge

Ideal Experiences:

• Building distributed training systems for AI models in HPC clusters
• Working with JAX/XLA for multi-host training
• Debugging and customizing third-party source code

Interview Process:

• After submitting your application, your CV and statement of exceptional work will be reviewed by the team.
• If selected, you will have a 15-minute phone interview with a team member asking basic questions.
• Successful candidates will proceed to the main process, which consists of:
• Four technical interviews including one-on-one research discussion & coding interviews (three total)
• A project deep dive to present your past exceptional work to a small audience
• We aim to complete the main process within one week, and all interviews will be conducted via Google Meet.

Annual Salary Range: $180,000 - $440,000 USD

California Consumer Privacy Act (CCPA) Notice