About this role

xAI is seeking an ML Infrastructure Engineer to build and optimize the high-performance ML platform powering recommendations on X. This role involves designing, building, and scaling GPU compute infrastructure, training frameworks, and data pipelines. You will collaborate with ML teams to productionize models and ensure the scalability and reliability of large-scale machine learning systems.

What we are looking for

6

Design, build, and scale GPU compute infrastructure
Develop training frameworks and experimentation tools
Develop and integrate large-scale data and ML systems
Collaborate with ML teams to productionize models
Ensure scalability, reliability, and efficiency of ML systems
Work across the full stack to solve complex problems

Skills mentioned

4

PythonC++RustMachine Learning