Defense AI/ML Engineer: Build Secure, Scalable Models

540; Arlington (VA); On-site; USD 120,000 - 190,000; Full time

Benefits offered by this job

Flexible PTO

Health insurance

401k plan

Professional development

Annual social events

Parking / metro reimbursement

Job summary

540 is seeking an AI/ML Engineer to support a mission-critical technology modernization effort for the Department of War. You will design, build, and maintain production AI/ML services and infrastructure that enable teams to develop, deploy, monitor, and scale models supporting complex defense missions.

The role involves collaborating with software engineers, data engineers, data scientists, cybersecurity teams, and mission stakeholders to create secure and reliable AI/ML systems that directly

Qualifications

• 4+ years of AI/ML engineering, software engineering, or data science experience.
• Experience developing and deploying production-grade AI or machine learning systems.
• Proficiency with Python and commonly used AI/ML frameworks.
• Experience building automated model training, validation, deployment, and monitoring pipelines.
• Experience with MLOps platforms, practices, and tools.

Responsibilities

• Design, build, and maintain AI/ML services, products, and lifecycle capabilities.
• Develop automated pipelines for model training, validation, testing, deployment, and monitoring.
• Create reusable frameworks, libraries, and shared components that accelerate AI/ML development.
• Build model-serving capabilities supporting secure, scalable, and reliable inference.
• Implement MLOps practices using CI/CD, infrastructure as code, automated testing, and source control.
• Develop model monitoring, performance tracking, drift detection, and operational health capabilities.
• Collaborate with data engineers and data scientists to prepare data and operationalize models.

Skills

AI/ML engineering

Python

CI/CD

Docker

Kubernetes

MLOps

Data pipelines

APIs

Cloud experience

Model deployment

Education

Bachelor's degree in CS/Engineering

Tools

AWS SageMaker

Kubeflow

Airflow

MLflow

Job description

540 is seeking an AI/ML Engineer to support a mission-critical technology modernization effort for the Department of War. You will design, build, and maintain production AI/ML services and infrastructure that enable teams to develop, deploy, monitor, and scale models supporting complex defense missions.

The role involves collaborating with software engineers, data engineers, data scientists, cybersecurity teams, and mission stakeholders to create secure and reliable AI/ML systems that directly