Senior Machine Learning Engineer

Rocket Lab; Tucson (AZ); On-site; USD 140,000 - 180,000; Full time

Job summary

Rocket Lab's Optical Systems division seeks a Senior Machine Learning Engineer II to design, train, and deploy deep learning models for electro-optical image processing at our Tucson site. You will build pipelines for data ingestion, feature engineering, and inference, collaborating with researchers and engineers on AI, ML, and computer vision tasks.

Requirements include a strong ML background, Python proficiency, experience with Keras/TensorFlow/PyTorch, and the ability to obtain a US

Qualifications

• Bachelor's degree and 8+ years experience in CS/engineering or higher degrees.
• Experience building deep learning models for computer vision or image processing.
• Proficiency in Python and ML libraries (Keras, TensorFlow, PyTorch).
• US citizenship and ability to obtain US government security clearance.

Responsibilities

• Develop ML models for EO/IR image processing.
• Train, deploy and maintain ML models for Optical Systems applications.
• Build ML pipelines for data ingestion, feature engineering, training and inference.
• Collaborate with researchers and engineers on AI, ML and computer vision.
• Perform rapid prototyping and integration into operational systems.
• Troubleshoot and maintain existing and new systems.
• Stay current with ML research and evaluate new tools and techniques.

Skills

Python

Machine learning

Deep learning

CI/CD

Version control

Testing

Containerization

Security clearance

Education

Bachelor's degree in CS/engineering

Master's degree in related field

Ph.D. in related field

Tools

Keras

TensorFlow

PyTorch

Job description

At Rocket Lab, we’re not just launching rockets — we're building the future of space. Our Space Systems team builds everything from complete spacecraft, precision payloads to the components and subsystems that allow them to thrive in space, like solar panels, flight software, star trackers, optical systems, separation systems, radios, and more.

Our Space Systems team has enabled more than 1,700 missions, ranging from interplanetary exploration, in-space manufacturing to national security and defense initiatives. The team has built spacecraft, payloads, and components for missions to the Moon and Mars, working with partners including NASA, the Space Development Agency, and the U.S. Space Force.

Whether it’s a single high-performance spacecraft, constellation, or the vertically integrated components that help them get to space — our world class Space Systems team is empowering some of the boldest and most ambitious space missions.

SENIOR MACHINE LEARNING ENGINEER II

Rocket Lab’s Optical Systems division solves mission-critical space domain and Intelligence, Surveillance, and Reconnaissance (ISR) challenges for Department of Defense (DoD) and Intelligence Community (IC) customers. Our vision is to revolutionize the space-based payload market with innovative and novel designs for space, terrestrial, and airborne environments. Building on more than 20 years of electro-optical and infrared systems innovation, Optical Systems delivers solutions to the warfighter for responsive, scalable sensing solutions across all orbital domains.

As a Senior Machine Learning Engineer II based at our Optical Systems sites in Tucson, AZ, you will have the opportunity to support Tranche 3 of the U.S. Space Development Agency's (SDA) Proliferated Warfighter Space Architecture (PWSA) and beyond by building deep learning neural networks for advanced Electro-Optical (EO/IR) image processing.

WHAT YOU’LL DO

• Contribute your experience to developing solutions for real-world problems.
• Design, train, and deploy machine learning models for Optical Systems applications
• Build and maintain ML pipelines for data ingestion, feature engineering, training, and inference.
• Collaborate with other researchers/engineers on artificial intelligence, machine learning, and computer vision.
• Preform rapid prototyping and enhanced development to be integrated into operational systems.
• Perform troubleshooting, bug fixes, and maintenance of existing and new systems.
• Stay current with ML research and evaluate new tools, frameworks, and techniques

WHAT WE’RE LOOKING FOR IN A SENIOR MACHINE LEARNING ENGINEER II:

• Bachelor's degree and 8+ years of experience, master’s degree and 6+ years of experience, or a Ph.D. in Computer Science, Electrical and Computer Engineering, Mechanical Engineering, Physics, or related field
• Strong background in machine learning including model selection, architecting, training, validation, testing, and deployment
• Experience in building deep learning neural network for computer vision, image processing, or video analysis (e.g., object detection, image segmentation, and tracking)
• Proficiency in Python
• Experience with deep learning libraries (Keras, TensorFlow, Pytorch, etc)
• Software engineering fundamentals: version control, testing, CI/CD, containerization
• U.S. citizenship is required, due to program requirements
• Ability to obtain and maintain an U.S. Government Security Clearance

NICE TO HAVE:

• Proficiency in C/C++/Rust
• Experience curating quality, real-world datasets for training deep learning models
• TS/SCI security clearance

This position may require prolonged periods of sitting, standing, walking, computer work, and occasional exposure to moderate levels of noise, dust, and fumes in production areas.