Machine Learning Research Engineer

Machine Learning Research Engineer

As an experienced engineer, you know that machine learning is critical to understanding and processing massive datasets. Your ability to conduct statistical analyses using machine learning (ML) techniques makes you an integral part of delivering customer‑focused solutions. In this role you’ll train, test, deploy, and maintain models that learn from data. You’ll own and define the direction of mission‑critical solutions using best‑fit ML algorithms and technologies.

You will work across self‑supervised pretraining, lab‑to‑scene alignment, multi‑task model training, uncertainty calibration, benchmarking, and release readiness. You’ll collaborate with data engineers, data scientists, solutions architects, and remote sensing scientists to build world‑class, stable, high‑performing PyTorch systems. You will bridge model research and production‑grade ML engineering and guide clients through the landscape of ML algorithms, tools, and frameworks.

• 4+ years of experience with ML engineering, research engineering, or applied ML development
• Experience with PyTorch, including building and training deep learning models
• Experience with transformer‑based models, self‑supervised learning, multi‑task learning, or large‑scale training pipelines
• Experience debugging model training issues such as instability, memory bottlenecks, data loader performance, and reproducibility
• Experience with software engineering fundamentals, including testing, code review, and maintainable ML workflows
• Active TS/SCI clearance and willingness to take a polygraph exam
• Bachelor’s degree in Computer Science, Machine Learning, Applied Mathematics, Physics, or Remote Sensing

• Experience with computer vision, scientific imaging, remote sensing, or hyperspectral data
• Experience with masked autoencoders, contrastive learning, retrieval models, or multimodal alignment
• Experience with uncertainty estimation, calibration, conformal prediction, or OOD detection
• Experience with distributed training, mixed precision, and GPU performance optimization
• Experience supporting model evaluation and qualification in high‑stakes or research‑heavy domains
• Master’s degree in a related field preferred;
  Doctorate a plus

Applicants selected will be subject to a security investigation and may need to meet eligibility requirements for access to classified information. TS/SCI clearance is required.

Projected salary range: $99,000.00 to $ (annualized USD). Benefits include health, life, disability, financial, and retirement plans, paid leave, professional development, tuition assistance, work‑life programs, and dependent care. Full‑time and part‑time employees working at least 20 hours a week are eligible for Booz Allen’s benefit programs.

All qualified applicants will receive consideration for employment without regard to disability, status as a protected veteran, or any other status protected by applicable federal, state, local, or international law.