Software Engineer - Multimodal Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

xAI · 5 months ago

Software Engineer - Multimodal Infrastructure

xAI is on a mission to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. They are seeking a Software Engineer specializing in Multimodal Infrastructure to develop and optimize scalable frameworks for large-scale machine learning tasks, including model training and data processing.

Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyMachine Learning
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Design, build, and implement large-scale distributed training systems
Profiling, debugging, and optimizing GPU utilization for multimodal model training and inference
Building scalable infrastructure for multimodal data management and crawling
Hardware / Software / Algorithm co-design

Qualification

PythonJax / PyTorchSpark / RayGPU optimizationRustPrioritization skillsCommunication skillsWork ethic

Required

Strong engineering skills with passion to improve different aspects of data and model
Has worked on one or more modalities other than text for model training/serving optimization
Keeping up with state-of-the-art techniques for preparing multimodal training data
Design, build, and implement large-scale distributed training systems
Profiling, debugging, and optimizing GPU utilization for multimodal model training and inference
Building scalable infrastructure for multimodal data management and crawling
Hardware / Software / Algorithm co-design
Strong communication skills

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

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

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

leader-logo
Toby Pohlen
Founding Member
linkedin
Company data provided by crunchbase