Job Description

xAI is seeking a highly motivated software infrastructure engineer to develop and optimize scalable frameworks for large-scale machine learning tasks, including model training, inference, and data processing, focusing on multimodal AI experiences beyond text.

Responsibilities

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

Requirements

• Strong engineering skills
• Experience with one or more modalities other than text
• Knowledge of state-of-the-art techniques for preparing multimodal training data
• Proficiency in Python, Jax / PyTorch, Spark / Ray, and Rust

Benefits

• No benefits