# Member of Technical Staff - Imagine Model

**Company:** [xAI](https://hotfix.jobs/companies/xai)
**Location:** Palo Alto, CA, Seattle, WA
**Role:** ML Engineering
**Skills:** PyTorch, Transformers, Computer Vision, Multimodal Ai, Reinforcement Learning, Image Generation, Video Generation, Data Pipelines, Inference Optimization, RLHF
**Posted:** 2026-03-20

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

## Job Description

## Responsibilities
- Create and drive engineering agendas to advance multimodal capabilities, with emphasis on image and video generation, editing, understanding, controllable/long-horizon synthesis, agentic planning, **RL training**, and world simulation (including audio integration for richer video experiences).
- Improve data quality through annotation, filtering, augmentation, synthetic generation, captioning, and in-depth data studies, particularly for visual and audio data.
- Design evaluation frameworks, metrics, benchmarks, evals, and reward models tailored to image/video/audio quality and coherence.
- Implement efficient algorithms for state-of-the-art model performance, including real-time inference, distillation, and scalable serving for visual content.
- Develop scalable data collection and processing pipelines for multimodal (primarily image/video-focused) datasets.
- Collaborate cross-functionally to integrate AI solutions into production and rapidly iterate based on user feedback.

## Basic Qualifications
- Track record in leading studies that significantly improve neural network capabilities and performance through better data or modeling.
- Experience in data-driven experiment designs, systematic analysis, and iterative model debugging.
- Experience developing or working with large-scale distributed machine learning systems.
- Ability to deliver optimal end-to-end user experiences.
- Hands-on contributor with initiative, excellence, strong work ethic, prioritization skills, and excellent communication.

## Preferred Skills and Experience
- Experience in **SFT**, **RL**, evals, human/synthetic data collection, or agentic systems.
- Proficiency in **Python**, **JAX/XLA**, **PyTorch**, **Rust/C++**, **Spark**, **Ray**, and related large-scale frameworks.
- Domain expertise in multimodal applications such as graphics engines, rendering techniques, image/video understanding and generation, world models, real-time simulation, or controllable/long-horizon visual content creation (audio/speech processing or music/audio generation experience is a plus where it supports video).
- Experience with agentic **RL** training, controllable/long-horizon generation, or multimodal agents that reason and act across modalities (especially in visual domains).

## Compensation and Benefits
**$180,000 - $440,000 USD**  
Base salary is just one part of our total rewards package at xAI, which also includes equity, comprehensive medical, vision, and dental coverage, access to a 401(k) retirement plan, short & long-term disability insurance, life insurance, and various other discounts and perks.

## Similar roles

- [Member of Technical Staff](https://hotfix.jobs/jobs/dd7c8bc4-741a-45d5-91a8-e6b1c4ad67e0) - xAI - Palo Alto, CA - $180k – $440k/yr
- [Staff Software Engineer, AI & App Experience](https://hotfix.jobs/jobs/78128221-8199-41cd-9c63-502d3f61d603) - Chime - New York, NY - $223k – $308k/yr
- [Staff + Senior Software Engineer, Cloud Inference Launch Engineering](https://hotfix.jobs/jobs/18b08a6a-5359-4836-b72e-c641fa718b04) - Anthropic - San Francisco, CA - $320k – $485k/yr
- [Senior Staff Software Engineer, Legal Automation](https://hotfix.jobs/jobs/7870d3e1-ebec-4f7b-9ae5-c62926a06d6e) - Coinbase - Remote - $254k – $299k/yr
- [Staff Machine Learning Engineer, Computer Vision](https://hotfix.jobs/jobs/2969c5c0-8479-43bb-b9b6-f8df93b528d7) - Pinterest - San Francisco, CA - $189k – $390k/yr

**Apply:** https://hotfix.jobs/jobs/6cef7dde-92b6-4b4e-a21b-b784988b4e56
**Canonical:** https://hotfix.jobs/jobs/6cef7dde-92b6-4b4e-a21b-b784988b4e56