Software Engineer - Voice Model

Job summary

Responsibilities

You'll join the Grok Voice Model team as a software engineer. You'll design, build, and run the software systems behind Grok Voice, which delivers natural, low-latency, multilingual spoken interactions across devices and in real time.

• Design, build, and maintain large-scale distributed data pipelines that ingest, process, and annotate real-world and synthetic audio, so the training and evaluation systems get…
• Build and optimize the distributed training infrastructure and software for speech-language models in JAX and PyTorch, including the systems that run supervised fine-tuning and…
• Build automated evaluation and testing systems that measure accuracy, audio quality, latency, and real-time performance, plus the experimentation infrastructure used to validate…
• Integrate voice models into production applications and real-time services, write the technical specifications for spoken interactions, and own deployment from prototype to…
• Profile and optimize system performance across training and inference, including throughput, latency, memory use, and compute efficiency.

What they're looking for

• Bachelor's degree or higher in Computer Science, Software Engineering, Computer Engineering, Electrical Engineering, or a related engineering field.
• Expert-level Python, writing clean, efficient, production-quality code for large-scale systems.
• Hands-on experience building distributed data processing systems with tools like Spark and Ray.
• Experience building and running large-scale distributed training and inference systems on Kubernetes.
• Experience implementing and optimizing model training and inference software in JAX or PyTorch.

Tech stack

• JAX
• PyTorch
• Python
• Spark
• Kubernetes