AI Researcher & Engineer - Multimodal (Image Gen)

The multimodal team at xAI creates magical AI experiences beyond text, enabling understanding and generation of content across various modalities, including image, video, and audio. As a multimodal researcher/engineer, you will drive the model’s multimodal capability through various aspects such as data, modeling, serving and product. You will collaborate with pre-training, post-training, and product teams to push the frontiers of model capability as well as the end-to-end user experience. Focus includes creating and driving research agenda to advance image and video generation and editing capabilities, improving data quality, developing data filtering/generation techniques, performing data study, creating evaluation frameworks and internal benchmarks, and designing and implementing effective and efficient algorithms for achieving state-of-the-art model performance. Ideal experiences include a track record in leading research that significantly improves the capability and performance of neural networks, experience in data-driven experiment designs and systematic analysis for iterative model debugging, experience in developing or working with large-scale distributed machine learning systems, and experience in graphics engines and rendering techniques is considered an advantage. The role is based in the Bay Area (San Francisco and Palo Alto), and candidates are expected to be located near the Bay Area or open to relocation. The tech stack includes Python, Jax, and Rust. The interview process involves an initial 15-minute phone interview followed by four technical interviews including research discussion, coding interviews, and a project deep-dive presentation. Benefits include base salary, equity, comprehensive medical, vision, and dental coverage, 401(k) retirement plan, disability insurance, life insurance, and various other discounts and perks.