AI/ML Engineer

Why 540?

540 is a forward-thinking company that the government turns to in order to #getshitdone. We don't just talk about innovation – we deliver it. We break down barriers, build impactful technology, and solve mission-critical problems.

About the Role

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. Working with software engineers, data engineers, data scientists, cybersecurity teams, and mission stakeholders, you will build reusable ML capabilities and automated pipelines using modern software engineering and MLOps practices.

Responsibilities

• Design, build, and maintain AI/ML services, products, and lifecycle capabilities supporting WDP
• Develop automated pipelines for model training, validation, testing, deployment, and monitoring
• Create reusable frameworks, libraries, and shared components that accelerate AI/ML development
• Deploy models in cloud-based or containerized environments
• Develop model-serving capabilities supporting secure, scalable, and reliable batch or real-time 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
• Support model explainability, reproducibility, governance, and lifecycle traceability
• Optimize AI/ML services and infrastructure for performance, scalability, reliability, and cost efficiency
• Collaborate with data engineers and data scientists to prepare data and operationalize models
• Partner with cybersecurity teams to implement security, access-control, auditing, and governance requirements
• Support troubleshooting issues spanning models, applications, data pipelines, infrastructure, and production services
• Document AI/ML architectures, engineering processes, and operational procedures

Requirements

• 4+ years of relevant AI/ML engineering, software engineering, or data science experience
• 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
• Experience deploying models in cloud-based or containerized environments
• Experience developing APIs, microservices, or model-serving capabilities for batch or real-time inference
• Understanding of model evaluation, performance monitoring, drift detection, explainability, and governance
• Experience with Docker, Kubernetes, or similar containerization and orchestration technologies
• Experience with CI/CD, infrastructure as code, automated testing, and source control
• Experience working within AWS, Azure, or Google Cloud
• Familiarity with data pipelines, feature engineering, distributed data processing, and data versioning
• Strong communication and collaboration skills, including the ability to document and explain technical decisions

Qualifications

• Bachelor's degree in Computer Science, Engineering, or a related technical field preferred; equivalent combinations of education and relevant experience will be considered

Skills

• Experience supporting DoW, federal, Advana, or other enterprise AI/ML and data platforms
• Experience with AWS SageMaker or comparable cloud AI/ML platforms
• Experience with MLflow, Kubeflow, Airflow, Argo Workflows, Ray, Feast, or similar tools
• Experience building AI/ML solutions in secure, regulated, classified, or mission-critical environments
• Familiarity with large language models, generative AI, retrieval-augmented generation, or foundation-model operations
• Experience implementing responsible AI, model-risk-management, or AI-governance practices

Benefits & Perks

• Flexible PTO + all Federal holidays off
• Health, dental and vision insurance plans
• Flexible Spending Account (FSA)
• 401k with employer match
• Company-sponsored life insurance, short- and long-term disability
• Professional development (training, certifications, conferences)
• Paid cloud developer accounts
• Referral bonuses
• HQ office perks (parking / metro reimbursement, nitro coffee & lunches)
• Annual social events (540 Week, hackathon, charity golf tournament, etc.)

EQUAL EMPLOYMENT OPPORTUNITY (EEO)

540's policy is to provide equal employment opportunity to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws. This policy applies to all terms and conditions of employment, including recruiting, hiring, placement, promotion, termination, layoff, recall, transfer, leaves of absence, compensation and training.