SpaceX

ML Engineer, Surrogate Modeling (Vehicle Engineering)

SpaceX
Apply
2 days ago
Hawthorne, CA, USAEntry Level
H1B Sponsor

Base Salary

$125k - $175k/yr

Responsibilities

  • Develop, train, evaluate, and deploy production-grade AI surrogate models for engineering simulation workflows.
  • Design neural architectures and training strategies for complex engineering domains.
  • Build scalable data pipelines for tens of thousands of high-fidelity simulation results.
  • Research and implement neural operators, physics-informed ML, and surrogate-modeling techniques.
  • Apply uncertainty quantification, active learning, adaptive sampling, and inverse-problem methods.
  • Collaborate on architecture, design, and code reviews with hardware, simulation, and domain engineers.
  • Validate AI systems for accuracy, robustness, and reliability in engineering applications.

Requirements

  • Bachelor’s degree in computer science, data science, engineering, mathematics, physics, or a related technical discipline, or 4+ years of professional software-development experience in lieu of a degree.
  • At least 1 year of Python software-development experience for machine learning, AI, or data-science applications.
  • Preferred: master’s or PhD focused on surrogate modeling or AI for scientific or engineering simulation.
  • Preferred: experience training, tuning, and deploying production-grade ML surrogate models in engineering workflows.
  • Preferred: expert knowledge of architectures such as Fourier Neural Operators, neural operators, MeshGraphNet, Transolver, graph neural networks, or physics-informed neural networks.
  • Preferred: experience with inverse problems, uncertainty quantification, active learning, adaptive sampling, and traditional simulation methods.
  • Preferred: proficiency with PyTorch, TensorFlow, or JAX and experience with surrogate-modeling libraries such as NVIDIA PhysicsNemo.
  • Preferred: Linux and GPU-accelerator development experience, software-engineering best practices, statistics, numerical methods, and core machine-learning algorithms.
  • Must be able to work extended hours and weekends as necessary.
  • Must satisfy applicable ITAR eligibility or authorization requirements.

Benefits

  • Base salary is listed across AI Software Engineer Level I and Level II ranges, with additional potential stock, stock options, long-term cash awards, discretionary bonuses, and employee stock purchase opportunities.
  • Benefits include medical, vision, dental, 401(k), disability and life insurance, paid parental leave, discounts, and other perks.
  • Employees may accrue three weeks of paid vacation, 10 or more paid holidays, and paid sick leave under company policy.
  • The role may require extended hours and weekend work as necessary.

Tech Stack

Categories

SpaceX

About SpaceX

1,001-5,000 employees

SpaceX designs, manufactures and launches the world’s most advanced rockets and spacecraft. The company was founded in 2002 by Elon Musk to revolutionize space transportation, with the ultimate goal of making life multiplanetary. SpaceX has gained worldwide attention for a series of historic milestones. It is the only private company ever to return a spacecraft from low-Earth orbit, which it first accomplished in December 2010. The company made history again in May 2012 when its Dragon spacecraft attached to the International Space Station, exchanged cargo payloads, and returned safely to Earth — a technically challenging feat previously accomplished only by governments. Since then Dragon has delivered cargo to and from the space station multiple times, providing regular cargo resupply missions for NASA. For more information, visit www.spacex.com.

Contact me