ML Infrastructure Engineer

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.

• 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