Epochs
HomeRoomsWatchATS ScoreJobsTopicsProgress
← All jobs

Model Engineer, Multimodal Understanding

xAI · Palo Alto, CA
✨ Reading the job post and pulling out the key points…
Full job post · as written by xAI

SpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today, SpaceX is actively developing the technologies to make this possible, with the ultimate goal of making life multiplanetary and expanding consciousness to the stars.

SpaceXSI is building Super Intelligence to help humanity understand the universe. Our teams are building the full Super Intelligence stack: from our Colossus supercomputers, to Grok models, to our suite of applications, including Grok Bot, Grok Build, Grok Imagine, Grok Voice, and X. We're creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next.

MODEL ENGINEER, MULTIMODAL UNDERSTANDING

You will join the multimodal team to push toward superhuman multimodal intelligence. Advance understanding and generation across modalities—image, video, audio, and text—spanning the full stack: data curation/acquisition, tokenizer training, large-scale pre-training, post-training/alignment, infrastructure/scaling, evaluation, tooling/demos, and end-to-end product experiences.

Collaborate cross-functionally with pre-training, post-training, reasoning, data, applied, and product teams to deliver frontier capabilities in multimodal reasoning, world modeling, tool use, agentic behaviors, and interactive human-AI collaboration. Contribute to building models that can see, hear, reason about, and interact with the world in real time at unprecedented levels.

RESPONSIBILITIES

  • Design, build, and optimize large-scale distributed systems for multimodal pre-training, post-training, inference, data processing, and tokenization at web/petabyte scale
  • Develop high-throughput pipelines for data acquisition, preprocessing, filtering, generation, decoding, loading, crawling, visualization, and management (images, videos, audio + text)
  • Advance multimodal capabilities including spatial-temporal compression, cross-modal alignment, world modeling, reasoning, emergent abilities, audio/image/video understanding & generation, real-time video processing, and noisy data handling
  • Drive data quality and studies: curation (human/synthetic), filtering techniques, analysis, and scalable pipelines to support trillion-parameter models
  • Create evaluation frameworks, internal benchmarks, reward models, and metrics that capture real-world usage, failure modes, interactive dynamics, and human-AI synergy
  • Innovate on algorithms, modeling approaches, hardware/software/algorithm co-design, and scaling paradigms for state-of-the-art performance
  • Build research tooling, user-friendly interfaces, prototypes/demos, full-stack applications, and enable rapid iteration based on feedback
  • Work across the stack (pre-training → SFT/RL/post-training) to enable reasoning, tool calling, agentic behaviors, orchestration, and seamless real-time interactions

BASIC QUALIFICATIONS

  • Experience writing production-quality code in Python, PyTorch, JAX, Rust, or C++

PREFERRED SKILLS AND EXPERIENCE

  • Expert-level proficiency in Python (core language), with strong experience in at least one of: JAX / PyTorch / XLA
  • Proven track record building or optimizing large-scale distributed ML systems (training/inference optimization, GPU utilization, multi-GPU/TPU setups, hardware co-design)
  • Deep experience designing and running data pipelines…
Apply on xAI's site ↗Opens xAI's official careers page. Your application goes straight to them.
Company
xAI
Location
Palo Alto, CA
Work mode
On-site
Get ready for this roleCheck your ATS scorePractise now
Epochs

Train like a model. Interview like a pro.

Practice

HomeTopicsAI interviewProgress

Together

RoomsWatchFeedback

Career

JobsATS Score
© 2026 EpochsFree to use. Sign in to save your progress across devices.
Apply on xAI's site ↗Opens xAI's official careers page. Your application goes straight to them.

Overview

Design and scale multimodal models and data pipelines for the multimodal team at xAI, delivering real‑time vision, audio, and language capabilities.

Skills

PythonPyTorchJAXRustC++Distributed trainingGPUTPUData pipelines

Responsibilities

  • Build and optimize distributed training/inference systems for multimodal data.
  • Develop high‑throughput pipelines for image, video, audio, and text processing.
  • Design data curation and filtering pipelines supporting trillion‑parameter models.
  • Create evaluation benchmarks and reward models reflecting real‑world usage.
  • Innovate algorithms and hardware‑software co‑design for state‑of‑the‑art performance.
  • Build research tools, demos, and full‑stack applications for rapid iteration.

Requirements

  • Write production‑grade Python code for ML frameworks.
  • Experience with PyTorch, JAX, Rust, or C++.
Nice to have
  • Expert Python with JAX or PyTorch
  • Built large‑scale distributed training systems
  • Designed high‑throughput multimodal data pipelines
  • Worked on hardware‑software co‑design for ML

About xAI

xAI develops large‑scale AI models and applications to advance human understanding of the universe.

What to prepare

Multimodal modelingDistributed systemsData pipelinesEvaluation benchmarks
Summarised by AI from the job post. Check the full post before applying.