jobs

ML Infrastructure Engineer

SpaceXAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team is small, highly motivated, and focused on engineering excellence. This organization is for indivi

Company

xAI

Location

Palo Alto, CA

Status

Open

Posted

2026-07-22T03:49:23+00:00

xAI is hiring an ML Infrastructure Engineer in Palo Alto to build and optimize the ML platform behind recommendations on X. Responsibilities include scaling GPU compute infrastructure, training frameworks and experimentation tools, developing data pipelines, integrating large-scale training and inference systems, and productionizing models with ML teams. The posting asks for a quantitative degree or equivalent experience, 2+ years in large-scale production or distributed systems, ML platform or training infrastructure work, Python plus C++ or Rust, and familiarity with JAX, PyTorch, CUDA and schedulers like Slurm. Base pay range is listed.