Senior Software Engineer (AI/ML)

540 is seeking a Senior Software Engineer (AI/ML) to support our partnership with Google and the Department of War in advancing mission-critical capabilities for a global data processing platform. This platform leverages Machine Learning, modern cloud and containerized technologies to ingest, process, and analyze high-volume, time-series data.

The system leverages cloud-native infrastructure, while incorporating AI/ML capabilities to enhance signal analysis, anomaly detection, and data-driven insights. Engineers on this team build and scale full-stack features across user interfaces, backend services, and data pipelines that power mission-critical analytics. You’ll play a key role in integrating AI/ML-driven capabilities into production systems, enabling faster, more accurate operational decision-making.

• Location : Patrick SFB, FL or Arlington, VA. Candidates should be local to either location. Onsite support may be required based on mission and customer needs. Travel up to 25%
• Citizenship & Clearance Requirement : Per client requirements, candidates must be U.S. Citizens with an active DoW clearance with TS/SCI eligibility
• Education Requirement: Bachelor's Degree in Computer Science or related field (preferred)
• 540 Internal Thrive Level: Senior Software 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.

HOW YOU’LL DRIVE IMPACT

Design and develop machine learning models and applications for deployment to cloud-native services

Collaborate with data engineers and data scientists to productionize machine learning models and data pipelines

Collaborate with a cross-functional team of architects, engineers, and scientists to design and deploy Machine Learning Operations (MLOps) at scale

Integrate AI/ML capabilities into production systems (e.g., model inference APIs, decision-support features, anomaly detection workflows)

Design and optimize data models and persistence layers to support both transactional and analytical workloads

Enable intelligent application behavior by delivering AI and ML features to user-facing applications

Contribute to technical design documentation and system architecture artifacts

Lead or participate in code reviews, testing, and troubleshooting to ensure high-quality software

Drive system reliability through robust testing

Improve system performance, scalability, and reliability as the platform evolves

REQUIRED SKILLS & EXPERIENCE

8+ years of professional software development experience

Experience developing microservices in containerized environments

Experience with, or a strong conceptual understanding of, LLM-based applications, RAG pipelines, and AI-driven decision systems

Experience with Python

Experience working with relational databases and data warehouses

Experience integrating AI/ML capabilities into applications (e.g., working with model inference APIs, ML pipelines, or data-driven features)

Ability to produce technical design documentation (system diagrams, architecture artifacts)

Familiarity with MLOps concepts (model deployment, monitoring, versioning)

Ability to design with future scalability and platform evolution in mind

Demonstrated ownership and ability to drive work from concept to production

Strong problem-solving skills and ability to navigate ambiguity

Excellent communication skills, including working directly with stakeholders or clients

NICE TO HAVE

Experience with Google Cloud Platform (GCP) or similar cloud environments

Experience designing and consuming RESTful APIs

Experience with API gateways and API management platforms, such as Google Apigee

Familiarity with API authentication and authorization (PKI, OAuth2, JWT, LDAP, SAML, etc.)

Experience working on U.S. Federal Government programs, particularly DoW environments

Google Cloud Professional-level certifications

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.)

Access to 540’s Washington Capitals & Nationals tickets

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.