AI Engineer

Booz Allen Hamilton • El Segundo, CA

4d • $99,000 - $225,000

About The Position

AI Engineer The Opportunity: As an experienced engineer, you know that machine learning is critical to understanding and processing massive datasets. Your ability to conduct statistical analyses on business processes using ML techniques makes you an integral part of delivering a customer-focused solution. We need your technical knowledge and desire to problem-solve to support innovation across the DoD market. As an Artificial Intelligence Engineer on our defense technology team, you’ll develop AI solutions that enhance the national security mission. In this role, you’ll own and define the direction of mission-critical solutions by implementing AI that enables faster mission success. You’ll be part of a large community of machine learning engineers across the company and collaborate with data engineers, data scientists, solutions architects, and product owners to deliver world-class solutions to implement a wide variety of AI use cases. Your skills and extensive technical expertise will guide clients as they navigate the landscape of ML algorithms, tools, and frameworks. Work with us to solve real-world challenges and define AI strategy for national security.

Requirements

• 4+ years of experience in database development, machine learning, data science, or data engineering
• 2+ years of experience supporting AI or ML workloads and data pipelines
• Experience in natural language processing, LLMs, transformers or Retrieval-Augmented Generation (RAG) architectures
• Experience with AWS cloud services and infrastructure
• Top Secret clearance
• Bachelor's degree

Nice To Haves

• 3+ years of experience with vector databases or graph databases
• Experience with specialized vector search optimization techniques
• Experience with real-time streaming and low-latency data systems
• Knowledge of graph neural networks and graph-based machine learning
• Top Secret clearance
• Master's degree
• AWS or Security+ Certification

Responsibilities

• Design and implement vector and graph database architectures to support AI model training, inference, and evaluation processes.
• Develop and maintain data pipelines for AI ingestion, processing, and analysis workflows in AWS cloud environments.
• Support continuous model training and evaluation processes through optimized data storage and retrieval systems.
• Ensure database solutions maintain explainability and auditability requirements for AI models.
• Implement and maintain database systems that support DoD AI ethical principles and compliance requirements.
• Design, develop, and optimize vector databases for embedding storage, similarity search, and AI model support.
• Build and maintain graph databases for complex relationship modeling and knowledge representation.
• Develop ETL or ELT pipelines for structured and unstructured data ingestion into vector and graph systems.
• Implement database monitoring, performance tuning, and scalability solutions.
• Create and maintain database documentation, schemas, and data governance protocols.

Benefits

• health, life, disability, financial, and retirement benefits, as well as paid leave, professional development, tuition assistance, work-life programs, and dependent care.
• recognition awards program acknowledges employees for exceptional performance and superior demonstration of our values.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level