ML Infrastructure Engineer: GPU Scale & Production Pipelines

SpaceX Super Intelligence; Palo Alto (CA); On-site; USD 180,000 - 440,000; Full time

Benefits offered by this job

Equity

Medical, Vision, and Dental coverage

401(k) retirement plan

Disability & Life Insurance

Employee discounts

Job summary

SpaceXSI is seeking an ML Infrastructure Engineer to design and optimize the high-performance ML platform powering recommendations on X. You will build scalable GPU compute infrastructure, training frameworks, and experimentation tools to accelerate rapid hypothesis testing.

You will collaborate with ML teams to productionize models, develop data pipelines, and ensure reliability across the stack while mentoring junior engineers and contributing to the growth of the team.

Qualifications

• Bachelor, Master, Post-graduate or PhD in computer science, machine learning, or other quantitative discipline; or equivalent work experience.
• 2+ years of industry experience with high traffic or large-scale production environments, distributed systems, GPU infrastructure, and/or deep learning applications
• 2+ years experience with ML platforms, training infrastructure, or collaboration with modeling engineers and data scientists
• Strong proficiency with Python and experience with compiled languages such as C++ or Rust

Responsibilities

• Designing, building, and scaling GPU compute infrastructure, training frameworks, and experimentation tools to enable rapid iteration on ML hypotheses
• Developing data pipelines and integrating large-scale data, training, and inference systems
• Collaborating with ML teams to productionize models and ensure seamless integration across the stack
• Ensuring scalability, reliability, and efficiency of large-scale machine learning systems
• Working across the full stack to solve complex problems independently
• Mentoring junior engineers and contributing to the growth of the team

Skills

Python

C++

Rust

ML platforms

Distributed systems

Linux systems

Education

BS/MS/PhD in CS/ML or related field

Tools

Slurm

Puppet/Ansible

Job description

SpaceXSI is seeking an ML Infrastructure Engineer to design and optimize the high-performance ML platform powering recommendations on X. You will build scalable GPU compute infrastructure, training frameworks, and experimentation tools to accelerate rapid hypothesis testing.

You will collaborate with ML teams to productionize models, develop data pipelines, and ensure reliability across the stack while mentoring junior engineers and contributing to the growth of the team.