Senior ML Engineer II — Space Optics & Computer Vision

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

Job summary

Rocket Lab in Tucson, AZ is seeking a Senior Machine Learning Engineer II to design, train, and deploy ML models for Optical Systems and support SDA's Proliferated Warfighter Space Architecture. You will build ML pipelines and collaborate with researchers to advance space-grade AI capabilities.

The role requires a strong ML and CV background, Python proficiency, and experience with Keras/TensorFlow/PyTorch; US citizenship and ability to obtain a security clearance are mandatory.

Qualifications

• Bachelor's degree and 8+ years of experience; master's or PhD in CS, ECE, ME, or physics.
• Strong background in ML including model selection, training, validation, testing, and deployment.
• Experience in building deep learning neural networks for computer vision, image processing, or video analysis.
• Strong math background, particularly linear algebra.
• Proficiency in Python.
• Experience with deep learning libraries (Keras, TensorFlow, PyTorch).
• Software engineering fundamentals: version control, testing, CI/CD, containerization.
• U.S. citizenship is required, due to program requirements.
• Ability to obtain and maintain a U.S. Government Security Clearance.

Responsibilities

• 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.
• Perform 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

Skills

Machine learning

Python

Deep learning

Computer vision

CI/CD

Software engineering

Education

Bachelor's degree in CS/EE/Physics or related

Master's or PhD in CS/EE/ME/Physics

Tools

Keras

TensorFlow

PyTorch

Git

Docker

Job description

Rocket Lab in Tucson, AZ is seeking a Senior Machine Learning Engineer II to design, train, and deploy ML models for Optical Systems and support SDA's Proliferated Warfighter Space Architecture. You will build ML pipelines and collaborate with researchers to advance space-grade AI capabilities.

The role requires a strong ML and CV background, Python proficiency, and experience with Keras/TensorFlow/PyTorch; US citizenship and ability to obtain a security clearance are mandatory.