Software Engineer - Voice Model

SpaceXAI; Palo Alto (CA); On-site; USD 150,000 - 450,000; Full time

Benefits offered by this job

Equity

Medical coverage

Vision coverage

Dental coverage

401(k) retirement plan

Disability insurance

Life insurance

Other perks

Job summary

SpaceXAI seeks exceptionally smart, execution-oriented engineers to help build the world’s best voice AI. You will work on the Grok Voice Model team, owning data curation, model pre-training, and evaluation to push quality, speed, and stability across languages and devices.

Join a small, hands-on team with a flat structure, where leadership comes from initiative and consistent delivery. The role promises a fast-paced, high-caliber environment and impactful voice experiences.

Qualifications

• Expert-level Python programming.
• Experience processing large-scale datasets with Spark and Ray.
• Proficiency in pre-training and post-training with JAX/PyTorch.
• Ability to design robust evaluation pipelines including AB tests.
• Experience with Kubernetes for distributed training and inference.

Responsibilities

• Design and operate large-scale speech data pipelines for model training.
• Advance pre-training and post-training processes with fine-tuning and RL.
• Develop evaluation frameworks covering metrics, human studies, content factuality.
• Collaborate with product teams to deploy voice models in real-time environments.

Skills

Python

Spark

Ray

JAX

PyTorch

Kubernetes

Distributed systems

Evaluation pipelines

Multilingual support

Proactive attitude

Tools

Kubernetes

Spark

Ray

PyTorch

JAX

Job description

SpaceXAI’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. 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 employees are expected to have strong communication skills. They should be able to concisely and accurately share knowledge with their teammates.

ABOUT THE ROLE:

You will join the Grok Voice Model team to help build the world’s best voice AI. We deliver smooth, natural, low-latency spoken interactions — expressive, multilingual, and reliable across devices and real-time scenarios. We own the full training pipeline: massive data curation, premium audio processing, frontier speech-language pre-training, and intensive post-training to push quality, speed, and stability to the limit.

Our goal: make talking to AI feel like conversing with the most charming, kind, and knowledgeable person imaginable. We’re seeking exceptionally smart, execution-oriented engineers to help us get there.

RESPONSIBILITIES:

• Design and execute large-scale speech data curation and processing pipelines, including collection of diverse real-world audio, synthetic data generation, and automated annotation workflows to enable high-quality model training and evaluation.
• Work on pre-training and post-training of speech-language models, with targeted enhancements through supervised fine-tuning, reinforcement learning, and other techniques to ensure Grok Voice responses are accurate, factually grounded, natural and idiomatic in spoken style, conversational in tone, and fluent across multiple languages.
• Build and iterate a comprehensive evaluation framework covering objective metrics (accuracy, quality, latency, expressiveness), human preference studies, content factuality assessments, real-time interaction quality, and experimentation infrastructure to measure and improve performance.
• Work closely with product teams to integrate voice models into applications and real-time environments, define spoken interaction specifications, and handle the full lifecycle from prototype to global-scale deployment for stable, low-latency, delightful voice experiences.

BASIC QUALIFICATIONS:

• Python expert with deep proficiency in writing clean, efficient code for AI/ML systems.
• Hands-on experience processing large-scale datasets using tools like Spark and Ray for cleaning, augmentation, and feature extraction.
• Proficiency in pre-training and post-training speech-language models using JAX/PyTorch, including supervised fine-tuning, reinforcement learning, and optimizations for accuracy, factuality, natural spoken style, detail, and multilingual fluency.
• Ability to set up and run rigorous evaluation pipelines: objective metrics, human preference studies, content factuality checks, and iterative A/B testing to drive model improvements.
• Experience building or working with large-scale distributed training and inference systems on Kubernetes.
• Proactive, self-driven attitude — ready to grind in a fast-paced, high-caliber team to deliver outstanding voice AI experiences.

COMPENSATION AND BENEFITS:

$150,000 - $450,000 USD

Base salary is just one part of our total rewards package at SpaceXAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.