About the role
-
The mid-training team at xAI aims to provide an omni model that can understand the universe through text, image, video, and audio. To accomplish this, we are looking for expert engineers in multimodal mid-training data
-
Design synthetic data algorithms that push the frontiers of model intelligence
-
Distill the intelligence of flagship models into flash models through synthetic data generation
-
Optimize mid-training data mixtures to boost the ceiling for RL
-
Engineer long-context data recipes
-
Develop robust and diverse evaluation for mid-training checkpoints
Tech Stack:
- Python
- JAX and XLA
- Spark
- Ray
Benefits
-
Health and wellness: Comprehensive health insurance including medical, dental, vision, and disability coverage
-
Life and family: Life and AD&D insurance and fertility benefits to ensure our team’s well-being and peace of mind
-
Flexible vacation: We work hard but avoid burnout. Take time off when you need it
-
Visa sponsorship: We support international talent with visa sponsorship to join our team
-
401(k) plan: Retirement savings plan to secure your financial future
-
The role is based in the Bay Area. Candidates are expected to be located near the Bay Area or open to relocation
-
Strong ability to design ML experiments
-
Expertise in ML and large model scaling, with familiarity across all kinds of scaling laws
-
Strong engineering abilities in Spark, Ray, and other frameworks for large-scale data processing
-
Familiarity with state-of-the-art techniques for curating AI training data for text, image, audio, and video modalities
-
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:
- Coding assessment in a language of your choice
- Technical sessions (2): These sessions will be testing your ability to formulate, design and solve concrete problems in real world with LLM
- Meet the Team: Present your past exceptional work and your vision with xAI to a small audience
-
Our goal is to finish the main process within one week. All interviews will be conducted via Google Meet
Millions of jobs, with real people getting hired every day
Questions, answered
Click "Apply with JobAssist" – we tailor your resume and application to this role and submit it for your approval.
Yes. This role at xAI was screened before publishing – we confirmed the employer before listing it.
The employer didn't disclose a salary range for this listing. JobAssist shows pay whenever it's available.
This position can be done from anywhere, with no in-office requirement.
Yes – every application is tailored from your profile and this job's requirements, and you can review and edit before it's sent.