Skip to content
xAIxAIPalo Alto, CA

Member of Technical Staff - Voice Model

Develop voice AI models for natural, low-latency spoken interactions on the Grok team. Handle data pipelines, model training with JAX/PyTorch, evaluations, and product integrations. Requires Python expertise, large-scale data processing, and distributed systems experience.

150k – 450k
On-siteML Engineering

About the role

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.
  • Work on pre-training and post-training of speech-language models, with targeted enhancements through supervised fine-tuning, reinforcement learning, and other techniques.
  • Build and iterate a comprehensive evaluation framework covering objective metrics, human preference studies, content factuality assessments, real-time interaction quality, and experimentation infrastructure.
  • 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.

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.
  • 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.

Compensation and Benefits

$150,000 - $450,000 USD base salary, plus equity, comprehensive medical, vision, dental coverage, 401(k), short & long-term disability insurance, life insurance, and various perks.

Skills

PythonSparkRayJAXPyTorchKubernetesSupervised Fine-TuningReinforcement LearningSpeech-Language ModelsData Curation

Similar roles

ML Engineering jobs
Sieve

Member of Technical Staff, Machine Learning

SieveSan Francisco, CA

Machine Learning Engineer owning the full ML lifecycle for multimodal video datasets at Sieve. Fine-tune VLMs, build evaluation/QA pipelines with frontier models, design filtering systems over internet-scale data, and ship production improvements for top AI labs. Requires strong Python, PyTorch, and production ML experience.

150k – 350k
On-site5+ YOEML Engineering
Modal

Member of Technical Staff

ModalNew York, NY +1

Member of Technical Staff conducting hands-on LLM inference research at Modal. Own end-to-end bets on techniques like speculative decoding, quantization, KV-cache management, and disaggregation to improve cost per token and tail latency on production workloads. Requires strong LLM serving stack expertise and a track record shipping research or systems.

150k – 350k
On-siteML Engineering
Parallel

Member of Technical Staff, Model Training

ParallelCalifornia

Own the training pipeline for search and agent models, building from product usage data through fine-tuning and evaluation to production deployment. Requires deep expertise in transformer fine-tuning, data curation, and training models for ranking, retrieval, and agent behavior.

150k – 300k
On-siteML Engineering
Parallel

Member of Technical Staff, Search Ranking

ParallelUnited States

Own the multi-stage ranking pipeline for web-scale search, balancing precision, recall, latency, and compute cost across retrieval, first-pass ranking, and neural reranking.

150k – 300k
On-site7+ YOEML Engineering
Second Nature

Staff Software Engineer, Engineering AI Team

Second NatureUnited States

Staff engineer builds AI-driven platform infrastructure for SDLC transformation, owns end-to-end experiments using AI agents like Claude, and ensures high-velocity code delivery with strong abstractions and real-world grounding. Requires staff-level architecture experience and AI-native workflows.

150k – 180k
RemoteML Engineering