ML Infrastructure Engineer
SpaceXAI - Palo Alto, CA
Hiring: ML Infrastructure Engineer Company: SpaceXAI Location: Palo Alto, CA Job Posted Time: 2026-09-10 10:23:39 Target Skills & Keywords : Ansible, C++, Data Pipeline, Deep Learning, Linux, Machine Learning, Puppet, PyTorch, Python, Rust About the job Experience: •Deep familiarity with modern ML frameworks such as JAX or PyTorch •Low-level understanding of compute systems, including distributed storage, NVIDIA drivers, CUDA toolkits, and networking •Comfortable with Linux systems and orchestration tools Required Skills: •As an ML Infrastructure Engineer, you'll have the opportunity to make a significant impact on the future of X and our work building advanced AI systems. •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 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 working 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 close collaboration with modeling engineers and data scientists •Strong proficiency with Python and experience with compiled languages such as C++ or Rust Compensation: •$180,000 - $440,000 / year Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!