Apply on Employer Site

xAI · 6 hours ago

Member of Technical Staff - Grok Imagine

Palo Alto, CA

Full-time

Onsite

Mid, Senior Level

$180K/yr - $440K/yr

xAI is seeking an engineer specializing in Multi-modal Post-Training to design and deploy advanced systems and algorithms for enhancing AI capabilities. The role involves developing scalable data collection systems, optimizing model performance, and collaborating with cross-functional teams to integrate AI solutions into production environments.

Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyMachine Learning

Growth Opportunities

H1B Sponsor Likely

Responsibilities

Develop scalable human and synthetic data collection systems for media data (images, video, audio)

Build infrastructure for large-scale image, video and audio data pipelines, curation, and annotation

Optimize inference and distillation for real-time generation

Collaborate with cross-functional teams to integrate advanced AI solutions into production environments

Create demos to explore the frontier of multimodal AI products and iterate quickly based on real-world feedback

Own the end-to-end experience of created applications, working with teams to improve user experience

Qualification

Multi-modal Post-TrainingMachine Learning SystemsPythonJAXRustEngineering SkillsInnovative Product DevelopmentProject Management

Required

Deep technical expertise with experience in multimodal post-training techniques

Capable of applying the latest advancements to improve model performance

Demonstrated success in developing scalable machine learning systems to fuel product expansion

Exceptional engineering skills, adaptable across various tech stacks

Self-motivated and able to independently drive complex projects to completion

Preferred

Proficiency in creating multimodal applications (e.g., images, video, audio) is a plus

Enthusiasm for crafting innovative consumer products using state-of-the-art technology

Benefits

Equity

Comprehensive medical, vision, and dental coverage

Access to a 401(k) retirement plan

Short & long-term disability insurance

Life insurance

Various other discounts and perks

Company

xAI

XAI is an artificial intelligence startup that develops AI solutions and tools to enhance reasoning and search capabilities.

Founded in 2023

Palo Alto, California, USA

11-50 employees

https://x.ai

H1B Sponsorship

xAI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (1)

Funding

Current Stage

Late Stage

Total Funding

$42.73B

Key Investors

Neptune Digital AssetsSpaceXMorgan Stanley

2026-01-06Series E· $20B

2025-12-11Secondary Market· $0.3M

2025-07-13Corporate Round· $5.32B

Leadership Team

Greg Yang

Co-Founder

Yuhuai Wu

Co-Founder

Recent News

IndiaTimes

Elon Musk says xAI is bringing changes to ‘For You’ tab on X: Here’s what’s new

2026-01-25

The Cool Down

New report uncovers disturbing findings about millions of photos generated by Elon Musk's chatbot: 'This is industrial-scale abuse'

2026-01-25

Tech Xplore

Grok created three million sexualized images, research says

2026-01-24

Company data provided by crunchbase