X

xAI

Generative AI chatbot leveraging real-time data

Model Engineer - Voice

Natural Language Processing & LLMsConfirmed live in the last 24 hours
Full-timePalo Alto, CA, USAIn-person

About the Job

Requirements

  • A bachelor's degree in computer science, software engineering, computer engineering, electrical engineering, or another engineering discipline is required.
  • Experience writing production-quality code in Python, PyTorch, JAX, Rust, or C++ is required.

Responsibilities

  • Design, build, and maintain large-scale distributed data pipelines that ingest, process, and annotate real-world and synthetic audio so that training and evaluation systems receive reliable, high-quality data.
  • Build and optimize distributed training infrastructure and software for speech-language models in JAX and PyTorch, including systems that run supervised fine-tuning and reinforcement learning at scale.
  • Build automated evaluation and testing systems that measure accuracy, audio quality, latency, and real-time performance, as well as experimentation infrastructure used to validate every model release.
  • Integrate voice models into production applications and real-time services, write technical specifications for spoken interactions, and own deployment from prototype to global scale with a focus on stability, low latency, and reliability.
  • Profile and optimize system performance across training and inference, including throughput, latency, memory use, and compute efficiency.

Desired Qualifications

  • Experience building and running large-scale distributed training and inference systems on Kubernetes.
  • Expert-level Python proficiency, including writing clean, efficient, production-quality code for large-scale systems.
  • Experience implementing and optimizing model training and inference software in JAX or PyTorch.
  • Experience building automated testing, evaluation, and A/B experimentation pipelines for production systems.
  • Hands-on experience building distributed data processing systems with tools such as Spark and Ray.

About the Company

xAI

Generative AI chatbot leveraging real-time data

xAI builds Grok, a generative AI chatbot with a Hitchhiker's Guide-inspired persona that uses real-time data from X to produce current, culturally aware responses. Grok is accessible to premium subscribers on X, via a standalone website, mobile apps, and through an API for developers. The company also develops large-scale AI infrastructure, such as the Colossus supercomputer, to train and run its models. Its goal is to pursue truth-seeking AGI research and grow a capital-intensive platform that could support a major public offering in the future.

Company size
1,001-5,000
Company stage
Series E
Total funding
$42.1B
Headquarters
Palo Alto, CA
Founded
2023

Recently Posted Jobs

Explore more opportunities at xAI

See more jobs