Multimodal AI Engineer (Image/Video) — Equity Eligible

Maxxd; Palo Alto, Northern (CA, KY); Hybrid; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Medical, Vision & Dental

401(k) plan

Job summary

SpaceXAI's Imagine Model Team is seeking a multimodal engineer to advance image, video, and audio understanding and generation. You will work on data curation, modeling, training, inference serving, and product integration across pretraining and post-training phases.

You will collaborate with product teams to push model frontiers, deliver end-to-end user experiences, and contribute hands-on across data, algorithms, and deployment in a fast-moving, highly collaborative environment.

Qualifications

• Experience leading neural network studies to significantly improve performance through better data or modeling.
• Experience designing data-driven experiments and systematic analysis with 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.

Responsibilities

• Create and drive engineering agendas to advance multimodal capabilities across image/video generation, editing, understanding, and synthesis.
• Improve data quality through annotation, filtering, augmentation, synthetic generation, captioning, and data studies.
• Design evaluation frameworks, metrics, benchmarks, evals, and reward models for image/video/audio quality and coherence.
• Implement efficient algorithms for state-of-the-art model performance, including real-time inference and scalable serving.
• Develop scalable data collection and processing pipelines for multimodal datasets (image/video-focused).
• Collaborate cross-functionally to integrate AI solutions into production and iterate based on user feedback.

Skills

Neural network research

Data-driven experiments

Distributed ML systems

Python

PyTorch

JAX/XLA

C++

Communication

Tools

Python

PyTorch

JAX/XLA

Rust/C++

Spark

Ray

Job description

SpaceXAI's Imagine Model Team is seeking a multimodal engineer to advance image, video, and audio understanding and generation. You will work on data curation, modeling, training, inference serving, and product integration across pretraining and post-training phases.

You will collaborate with product teams to push model frontiers, deliver end-to-end user experiences, and contribute hands-on across data, algorithms, and deployment in a fast-moving, highly collaborative environment.