Machine Learning/AI Engineer

Posted Yesterday
Hiring Remotely in United States
Remote
135K-145K Annually
Mid level
Information Technology • Consulting
The Role
Develop and deploy production-grade AI/ML models, design end-to-end pipelines, and ensure compliance while collaborating with stakeholders in a federal environment.
Summary Generated by Built In

Who is Element?

We serve as a partner at the intersection of innovation and our clients' needs, efficiently crafting meaningful user experiences for government and commercial customers. By breaking down complex problems to their fundamental elements, we create modern digital solutions that drive efficiencies, maximize taxpayer dollars, and deliver essential outcomes that serve the people. 

Why Work at Element?

Make an impact that resonates-join our vibrant team and discover how you can improve lives through digital transformation.  Our talented professionals bring unparalleled energy engagement, setting a higher standard for impactful work. Come be a part of our team and shape a better future. 

Position Summary:

We are seeking a highly skilled and motivated Machine Learning / AI Engineer to support a federal government program focused on designing, developing, and deploying production-grade artificial intelligence and machine learning solutions. The role will focus on building scalable, secure, and high-performing AI models that support mission-critical decision-making and operational efficiency.

The Machine Learning/AI Engineer will work across the full model lifecycle—from data preparation and feature engineering to model training, deployment, and monitoring—in a regulated government environment.

Key Responsibilities

  • Develop and deploy at least three (3) production-grade AI/ML models, including use cases such as predictive maintenance, anomaly detection, classification, forecasting, or optimization.
  • Design, build, and maintain end-to-end machine learning pipelines, including data preprocessing, feature engineering, model training, evaluation, and deployment.
  • Apply deep expertise in PyTorch and/or TensorFlow to develop and fine-tune advanced machine learning and deep learning models.
  • Implement and support scalable model serving architectures, ensuring high availability, low latency, and secure inference in production environments.
  • Collaborate with data engineers to access, transform, and prepare large-scale datasets for model training and inference.
  • Partner with product owners, analysts, and stakeholders to translate business requirements into machine learning solutions.
  • Monitor model performance in production, including drift detection, accuracy tracking, and retraining strategies.
  • Ensure compliance with federal security, privacy, and governance standards in all AI/ML implementations.
  • Participate in Agile development cycles, including sprint planning, design reviews, and technical demonstrations.
  • Document model architectures, training methodologies, and deployment processes for maintainability and auditability. 

Minimum Requirements

  • Bachelor’s or Master’s degree in Computer Science, Data Science, Artificial Intelligence, Engineering, Mathematics, or related field (or equivalent experience).
  • 3+ years of experience in machine learning, data science, or AI engineering roles.
  • Proven experience delivering production-grade machine learning models in real-world environments.
  • Expert proficiency in Python and ML frameworks (scikit-learn, TensorFlow, PyTorch).
  • Advanced SQL development experience, including complex queries, performance tuning, and data transformation logic.
  • Experience leveraging Large Language Models (LLMs) in development.
  • Understanding of core concepts such as context windows, prompt design, and input/output structures, with the ability to apply AI tools effectively in building and enhancing solutions.
  • Experience building and deploying models using scalable serving frameworks (e.g., REST APIs, containerized deployments, or cloud-based inference services).
  • Experience working with large-scale structured and unstructured datasets.
  • Strong understanding of machine learning concepts including supervised/unsupervised learning, deep learning, model evaluation, and feature engineering.
  • Experience working in the federal government or other highly regulated environments with security and compliance requirements.
  • Strong analytical and problem-solving skills.
  • Ability to communicate complex technical concepts to non-technical stakeholders.
  • Strong collaboration skills across engineering, data, and product teams.
  • Ability to work independently in a fast-paced, mission-driven environment.
  • High attention to detail with a focus on model reliability and production readiness.
  • US Citizenship or Permanent Residency required.
  • Must reside in the Continental US.
  • Depending on the government agency, specific requirements may include public trust background check or security clearance.

Preferred Qualifications

  • Experience deploying models in cloud environments (AWS, Azure, or GCP).
  • Familiarity with MLOps tools and frameworks (e.g., MLflow, Kubeflow, SageMaker, Vertex AI).
  • Experience with streaming or real-time inference systems.
  • Background in predictive maintenance, anomaly detection, or operational analytics use cases.
  • Familiarity with Docker, Kubernetes, and CI/CD pipelines for machine learning systems.
  • Experience supporting healthcare (e.g., CMS), or other federal mission systems. 

Our People
We invest in the lives of our employees, both in and out of the workplace, by providing competitive pay and benefits packages. Benefits offered may include health care, dental, vision, life insurance; 401(k); paid time off including PTO, holidays, and any other paid leave required by law. 
 
Location
Be in your Element. We are a remote first company based out of Washington, DC.
 
Element is an equal opportunity employer  All qualified applicants will receive consideration for employment without regard to age, ancestry, race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or veteran status, marital status, protected veteran status, or any other legally protected class. 
 
We believe in a world where solutions we build improve the lives of those who use them.

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
Washington, District of Columbia
32 Employees
Year Founded: 2009

What We Do

We are your human-centric innovation partner, crafting impactful user experiences for government and commercial sectors with an unwavering commitment to efficiency and excellence. Our Core Capabilities: CMS Data Experience Our team has established deep domain experience in insurance, claims, and quality data throughout the CMS ecosystem. We equip healthcare stakeholders with insights to make informed policy decisions through strategic data management, integrating diverse data sources, and conducting analysis. Enabling Interoperability in Healthcare We build functionality using standards such as FHIR to allow for data interoperability between government agencies, providers, and patients. Guided by our long-time partnership with CMS and healthcare stakeholders, we advocate for patient data accessibility, striving to enhance the quality of care delivered. Leaders in DevSecOps As some of the earliest practitioners of DevSecOps within CMS, we exemplify the significance of a carefully engineered CI/CD pipeline. Amidst unprecedented HealthCare.gov registrations, our team safeguards its stability and reliability. Consulting to Innovate At Element, we act as consultative partners at the crossroads of innovation and our clients' needs. No matter the challenge, we identify and prioritize the appropriate methods, ensuring tailored solutions and using modern technologies. We believe in a world where solutions we build improve the lives of those who use them, and the elements of Innovation, Collaboration, Integrity, Equity, Creativity are the foundation

Similar Jobs

General Motors Logo General Motors

Machine Learning Engineer

Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
Remote or Hybrid
District of Columbia, USA
165000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office or Remote
Basking Ridge, NJ, USA
160000 Employees
73K-130K Annually

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office or Remote
Eden Prairie, MN, USA
160000 Employees
92K-164K Annually

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office or Remote
Raleigh, NC, USA
160000 Employees
113K-193K Annually

Similar Companies Hiring

Amplify Platform Thumbnail
Fintech • Financial Services • Consulting • Cloud • Business Intelligence • Big Data Analytics
Scottsdale, AZ
62 Employees
Standard Template Labs Thumbnail
Artificial Intelligence • Information Technology • Software
New York, NY
25 Employees
Golden Pet Brands Thumbnail
Digital Media • eCommerce • Information Technology • Marketing Tech • Pet • Retail • Social Media
El Segundo, California
178 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account