ML Infrastructure Engineer

What this role involves

xAI is hiring an ML Infrastructure Engineer to build and optimize the ML platform powering recommendations on X. The role involves designing GPU compute infrastructure, developing data pipelines, and collaborating with ML teams to productionize models. Candidates need 2+ years of industry experience with large-scale production environments and ML platforms.

Skills this posting asks for

• python
• c++
• rust
• jax
• pytorch
• nvidia drivers
• cuda
• linux
• slurm
• puppet
• ansible
• gpu infrastructure
• distributed systems
• ml platforms
• training infrastructure
• data pipelines
• inference systems

Requirements

• 2 years of experience
• Level: mid

From the employer’s posting

SpaceXAI’s mission is to create AI systems that can accurately understand the universe and aid humanity in its pursuit of knowledge. Our team i…