Principal Machine Learning Engineer (EO/IR Image Analysis)
Job Description
Rocket Lab is hiring a Principal Machine Learning Engineer to support advanced EO/IR image processing efforts for Optical Systems in Chantilly, VA.
Responsibilities
- Apply experience to develop solutions for real-world problems
- Design, train, and deploy machine learning models for Optical Systems applications
- Build and maintain machine learning pipelines covering data ingestion, feature engineering, training, and inference
- Partner with researchers and engineers working across artificial intelligence, machine learning, and computer vision
- Conduct rapid prototyping and enhanced development for integration into operational systems
- Handle troubleshooting, bug fixes, and maintenance for existing and newly developed systems
- Stay current with machine learning research and assess new tools, frameworks, and techniques
Requirements
- Bachelor’s degree plus 12+ years of experience, or a master’s degree or Ph.D. in Computer Science, Electrical and Computer Engineering, Mechanical Engineering, Physics, or a related field
- Strong machine learning background across model selection, architecture, training, validation, testing, and deployment
- Experience building deep learning neural networks for computer vision, image processing, or video analysis (object detection, image segmentation, and tracking)
- Strong math foundation, especially linear algebra
- Proficiency in Python
- Experience with deep learning libraries including Keras, TensorFlow, and PyTorch
- Software engineering fundamentals including version control, testing, CI/CD, and containerization
- U.S. citizenship required due to program requirements
- Ability to obtain and maintain a U.S. Government Security Clearance
Technologies
- Python
- Keras
- TensorFlow
- PyTorch
Location and Work Setting
- Chantilly, VA
- Onsite
Compensation
- USD 198,900 - 318,200 per year
Security Clearance (Nice to Have)
- TS/SCI security clearance
Additional Nice to Have
- Proficiency in C/C++/Rust
- Experience curating high-quality real-world datasets for training deep learning models