Latest ML Engineering jobs at xAI

Build and own a mission-critical research platform for evaluating AI model capabilities and behaviors at xAI. Design instruments, datasets, grading schemes, and infrastructure to measure, diagnose, and improve models while shipping delightful internal tools.

Build and scale GPU compute infrastructure, training frameworks, data pipelines, and ML platforms to power large-scale training and inference at xAI. Requires strong distributed systems and ML infrastructure experience plus Python proficiency.

Build and optimize the RL training framework and infrastructure for large-scale workloads at SpaceXAI, from ablations to production runs. Requires experience with distributed systems and proficiency in Python, JAX, Rust, or C++.

Build and optimize the high-performance inference platform serving Grok at massive scale. Design distributed serving infrastructure, low-level GPU optimizations, quantization, speculative decoding, and CI/CD for production reliability and low latency.

Engineer optimizing RL inference stack for workloads from ablations to production training. Requires experience with large-scale distributed systems, LLM inference, and proficiency in Python/C++/Rust with PyTorch/JAX/CUDA.

Develops and operates large-scale search engine infrastructure, including retrieval algorithms, indexing, and ML ranking models integrated with Grok AI. Requires experience with search systems, vector databases, and production ML in Python, Go, or Rust.

Develops advanced post-training and reinforcement learning techniques like RLHF/DPO and reward modeling to enhance AI model reasoning, truthfulness, and real-world capabilities at xAI. Seeks passionate AI enthusiasts obsessed with truth-seeking models; prior experience preferred but not required.

Develops large-scale distributed systems and pipelines for multimodal AI pre-training, post-training, and inference across image, video, audio, and text. Requires expert Python proficiency, experience with JAX/PyTorch/XLA, and scaling multimodal ML systems.

Develops and trains advanced AI models to tackle critical challenges in truth-seeking AI systems. Requires deep passion for building useful models and power user experience with AI; prior large-scale training experience preferred.

Develops multimodal AI models focused on image, video, and audio for high-fidelity generation, understanding, and agentic systems. Drives data curation, training, evaluation, and production integration of cutting-edge models.

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.

Develop and integrate Grok AI models into the Ads Platform to power bidding, auction, ranking, prediction, and optimization systems at massive scale. Requires 3+ years experience with large-scale AI-powered advertising products.

Develop and scale recommendation algorithms, ranking, and search systems using AI stacks and data infrastructure for a platform with 600M users. Requires expertise in recommender systems, deep learning frameworks like JAX/PyTorch, and tools like Kafka/Spark.

Develops multimodal mid-training data pipelines for omni models handling text, image, video, and audio. Requires ML expertise, scaling laws knowledge, experiment design, and strong engineering in large-scale data frameworks like Spark and Ray.