Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Sr. Model Engineer, Imagine

Home > Python programming jobs

Sr. Model Engineer, Imagine in USA new

  • Palo Alto, CA

Responsibilities

  • Advance multimodal capabilities in image and video generation, editing, understanding, controllable and long-horizon synthesis, agentic planning, RL training, world simulation, and audio-enhanced video.
  • Improve visual and audio data quality through annotation, filtering, augmentation, synthetic generation, captioning, and data studies.
  • Design evaluation frameworks, metrics, benchmarks, evaluations, and reward models for image, video, and audio quality and coherence.
  • Implement efficient algorithms, real-time inference, model distillation, and scalable serving for visual content.
  • Develop scalable data collection and processing pipelines for multimodal datasets.
  • Collaborate with product teams to integrate AI solutions into production and iterate based on user feedback.

Requirements

  • Production-quality coding experience in Python, PyTorch, JAX, Rust, or C++.
  • Experience with multimodal applications such as graphics engines, rendering techniques, image or video understanding and generation, world models, real-time simulation, or controllable and long-horizon visual content creation.
  • Preferred experience in audio or speech processing, music or audio generation, data-driven experiment design, systematic analysis, model debugging, large-scale distributed machine learning, or improving neural network capabilities through data and modeling.
  • Preferred experience with SFT, RL, evaluations, human or synthetic data collection, agentic systems, agentic RL training, controllable or long-horizon generation, or multimodal agents.
  • Proficiency in Python, JAX/XLA, PyTorch, Rust/C++, Spark, Ray, and related large-scale frameworks.
  • Ability to contribute hands-on, prioritize effectively, communicate well, and deliver end-to-end user experiences.

Benefits

  • $220,000 - $357,000 USD base salary.
  • Equity, comprehensive medical, vision, and dental coverage, and access to a 401(k) retirement plan.
  • Short- and long-term disability insurance, life insurance, discounts, and other perks.

Salary: $220k - $357k/yr

xAI

xAI builds advanced AI systems designed to deeply understand the universe and support humanity's quest for knowledge. Targeting innovators and thinkers, our platform emphasizes engineering excellence and hands-on collaboration, fostering an environment where curiosity drives results.

Similar positions

Sr. Staff Software Engineer

  • Pinterest
  • Full time
  • USA
  • 11/08/2026
  • Salary: $209k-$429k
  • Remote

Software Engineer, AI Automation

  • Coalition, Inc.
  • Full time
  • USA
  • 11/08/2026
  • Salary: $115k-$168k
  • Remote

Staff Machine Learning Engineer, Consumer

  • Reddit
  • Full time
  • USA
  • 11/07/2026
  • Salary: $230k-$322k
  • Remote

Software Engineer, AI Automation

  • Coalition, Inc.
  • Full time
  • Canada
  • 11/07/2026
  • Salary: C$116k-C$161k
  • Remote

Sr. Software Integration Engineer

  • Ultima Genomics
  • Full time
  • Remote
  • 11/07/2026
  • Salary: $120k-$160k
  • United States