AI Engineer & Researcher, xAI
Pretraining and Scaling
$180-440k
Building artificial intelligence
Open for applications
Building artificial intelligence
21-100 employees
Open for applications
$180-440k
21-100 employees
Company mission
To build AI systems that can help humanity understand the world better.
Role
Who you are
- This organization is for individuals who appreciate challenging themselves and thrive on curiosity
- All employees are expected to be hands-on and to contribute directly to the company’s mission
- 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
- Building distributed training systems for AI models in HPC clusters
- Configuring and troubleshooting hardware and operating-systems for maximum performance
- Working with JAX/XLA for multi-host training
- Writing custom CUDA kernels in either C++ or via Triton
What the job involves
- 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."
- Training trillion parameter neural networks at scale, as well as a variety of smaller specialized models
- Profiling, debugging, and optimizing multi-host GPU utilization
- Improving and innovating inference stack to serve millions of users
- 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
Application process
- 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:
- Coding assessment in a language of your choice
- Systems hands-on: Demonstrate practical skills in a live problem-solving session
- Project deep-dive: Present your past exceptional work to a small audience
- Meet and greet with the wider team
- Our goal is to finish the main process within one week
- All interviews will be conducted via Google Meet
Our take
xAI is dedicated to tackling the limitations of current AI systems by developing Grok, an AI model designed to provide accurate, reliable, and witty responses to a wide range of queries. Grok is equipped with real-time knowledge access, allowing it to offer insightful answers and even suggest relevant questions.
The company is focused on enhancing Grok's reasoning abilities, reliability, and safety through research directions such as scalable oversight mechanisms, formal verification techniques, and adversarial robustness improvements. Additionally, xAI aims to broaden Grok's applicability by integrating multimodal capabilities like vision and audio, enabling real-time interactions and assistance in various domains.
As xAI continues to refine Grok's capabilities, the company envisions a future where AI serves as a powerful tool for advancing scientific understanding, fostering innovation, and contributing to societal well-being.
Steph
Company Specialist at Welcome to the Jungle
Insights
Few candidates hear
back within 2 weeks
Company
Funding (last 2 of 3 rounds)
Nov 2024
$5bn
LATE VC
May 2024
$6bn
SERIES B
Total funding: $11.1bn
Company benefits
- Medical, vision, dental
- STD, LTD and life insurance
Company HQ
Downtown, Burlingame, CA
Articles
Leadership
Elon Musk
(Founder)Founder, chairman, CEO, and CTO of SpaceX. CEO and product architect of Tesla. Chairman and CTO of X Corp, founder of the Boring Company, co-founder of Neuralink and OpenAI, and president of the Musk Foundation.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 30 more jobs at xAI