xAI Logo

xAI

AI Engineer & Researcher - Multimodal

Posted 5 Days Ago
2 Locations
Entry level
2 Locations
Entry level
The AI Engineer & Researcher at xAI will focus on training neural networks at scale, mastering all data modalities, implementing state-of-the-art methods, and innovating new ideas in the field of AI. Ideal candidates will have experience building distributed training systems, working with JAX/XLA, and debugging third-party source code.
The summary above was generated by AI

About xAI

xAI’s 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 motivated, and focused on engineering excellence. This organization is for individuals who appreciate challenging themselves and thrive on curiosity. Engineers are encouraged to work across multiple areas of the company, and as a result, all engineers and researchers share the title "Member of Technical Staff."

We operate with a flat organizational structure. All employees are expected to be hands-on and to contribute directly to the company’s mission. Leadership is given to those who show initiative and consistently deliver excellence. Work ethic and strong prioritization skills are important.

All engineers and researchers are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

xAI does not have recruiters. Every application is reviewed directly by a technical member of the team.

Tech Stack

  • Python
  • JAX and XLA
  • Rust

Location

The role is based in the Bay Area [San Francisco and Palo Alto]. Candidates are expected to be located near the Bay Area or open to relocation.

Focus

  • Training trillion parameter neural networks at scale, as well as 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 the deep learning literature.
  • Innovating new ideas that bring us closer to our goal: Developing AI systems that can accurately understand the universe and create new knowledge.

Ideal Experiences

  • Building distributed training systems for AI models in HPC clusters.
  • Working with JAX/XLA for multi-host training.
  • Digging into third-party source code for debugging and customization.

Interview Process

After submitting your application, the team reviews your CV and statement of exceptional work. If your application passes this stage, you will be invited to a 15 minute interview (“phone interview”) during which a member of our team will ask some basic questions. If you clear the initial phone interview, you will enter the main process, which consists of four technical interviews:

  1. One on one research discussion & coding interviews  (three meetings total)
  2. Project deep-dive: Present your past exceptional work to a small audience

Our goal is to finish the main process within one week. We don’t rely on recruiters for assessments. Every application is reviewed by a member of our technical team. All interviews will be conducted via Google Meet.

Annual Salary Range

$180,000 - $440,000 USD

California Consumer Privacy Act (CCPA) Notice

Top Skills

Python
Rust

Similar Jobs

Be an Early Applicant
2 Days Ago
San Francisco, CA, USA
Hybrid
19 Employees
Senior level
19 Employees
Senior level
Software
The Senior AI Research Engineer will develop and launch generative AI systems, create new models for 3D applications, maintain training scripts, and collaborate with cross-functional teams. The role requires expertise in AI research, model deployment, and clear communication of concepts to diverse stakeholders.
Be an Early Applicant
3 Days Ago
Sunnyvale, CA, USA
166,834 Employees
Senior level
166,834 Employees
Senior level
Internet of Things
Conduct research and engineering in AI and machine learning for embodied AI in autonomous driving and robotics. Collaborate globally to implement cutting-edge research findings into Bosch's operational frameworks, ensuring high-quality integration while staying updated with the latest technologies and document research through publications and patents.
Be an Early Applicant
4 Days Ago
Mountain View, CA, USA
1,218 Employees
Mid level
1,218 Employees
Mid level
Artificial Intelligence
As an AI for Code Research Scientist, you will improve models and research for real-world problems by prototyping concepts and running experiments, collaborating with a team to drive foundational research direction.

What you need to know about the Toronto Tech Scene

Although home to some of the biggest names in tech, including Google, Microsoft and Amazon, Toronto has established itself as one of the largest startup ecosystems in the world. And with over 2,000 startups — more than 30 percent of the country's total startups — Toronto continues to attract new businesses. Be it helping entrepreneurs manage their finances, simplifying business operations by automating payroll or assisting pharmaceutical companies in launching new drugs, the city's tech scene is just getting started.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account