For companies
  • Hire developers
  • Hire designers
  • Hire marketers
  • Hire product managers
  • Hire project managers
  • Hire assistants
  • How Arc works
  • How much can you save?
  • Case studies
  • Pricing
    • Remote dev salary explorer
    • Freelance developer rate explorer
    • Job description templates
    • Interview questions
    • Remote work FAQs
    • Team bonding playbooks
    • Employer blog
For talent
  • Overview
  • Remote jobs
  • Remote companies
    • Resume builder and guide
    • Talent career blog
Shield AI
Shield AI

AI/ML Data Engineer

Location

Remote restrictions apply
See all remote locations

Salary Estimate

N/AIconOpenNewWindows

Seniority

N/A

Tech stacks

Amazon
Data
AI
+28

Visa

U.S. visa required

Permanent role
a day ago
Apply now

Job ID 2510286

Location REMOTE WORK, VA, US

Date Posted 2025-09-25

Category Engineering and Sciences

Subcategory Solutions Archt

Schedule Full-time

Shift Day Job

Travel No

Minimum Clearance Required None

Clearance Level Must Be Able to Obtain Public Trust

Potential for Remote Work Yes

Description

We are seeking a Data Engineer with hands-on AI/ML and LLM project experience in AWS to join the AWS AI/GenAI Solutions Team within the IRS Advanced Analytics Program (AAP). This role is responsible for designing and maintaining data pipelines and feature engineering workflows that directly enable LLM/GenAI and AI/ML model development, training, and deployment on AWS services such as SageMaker and Bedrock.

As part of the AAP common services mission, the Data Engineer will deliver secure, scalable, and reusable AWS-native data engineering solutions that simplify onboarding for IRS mission teams. The ideal candidate combines expertise in AWS data services with AI/ML-focused engineering to ensure mission teams can build and operationalize models efficiently.

Key Responsibilities

  • Design, build, and optimize data pipelines in AWS (Glue, Lambda, Step Functions, S3, RDS, Redshift) to support AI/ML and LLM workloads.
  • Implement data ingestion, transformation, and feature engineering workflows that feed SageMaker and Bedrock models.
  • Collaborate with mission data scientists to ensure datasets are structured and optimized for LLM fine-tuning, inference, and prompt engineering.
  • Integrate pipelines into CI/CD workflows for automated, repeatable, and compliant model operations.
  • Apply security and governance controls (IAM roles, encryption, audit logging) to protect sensitive IRS data.
  • Develop and maintain data validation, schema enforcement, and monitoring routines to ensure reliability and compliance.
  • Work with MLOps/SRE engineers to align pipelines with model lifecycle operations (staging, promotion, retraining).
  • Partner with Product Manager and Chief Architect to align AWS data engineering capabilities with AAP roadmap milestones.

Qualifications

Required Qualifications

  • Bachelor’s degree in computer science, Data Engineering, or related field.
  • 10+ years of data engineering experience on AWS, including AI/ML-focused use cases.
  • Hands-on expertise with AWS data services (Glue, Lambda, S3, Redshift, RDS, Step Functions).
  • Strong proficiency in Python, SQL, and data transformation frameworks.
  • Experience delivering feature engineering and data prep for SageMaker/Bedrock model development.
  • Familiarity with CI/CD integration and IaC (Terraform, CloudFormation).
  • Awareness of AI/ML lifecycle data needs (training, fine-tuning, inference, retraining).

Desired Skills

  • Certifications AWS Certified Data Analytics Specialty, AWS Certified Machine Learning Specialty, or Solutions Architect Associate/Professional.
  • Experience working with LLM-specific pipelines (prompt data preparation, response validation, fine-tuning datasets).
  • Familiarity with federal compliance frameworks (FedRAMP, NIST 800-53) and embedding compliance into AWS data workflows.
  • Exposure to Trustworthy AI practices (bias detection, data lineage, explainability).
  • Strong collaboration skills to work across architects, AI/LLM engineers, and mission data scientists.

Target salary range $160,001 - $200,000. The estimate displayed represents the typical salary range for this position based on experience and other factors.

About Shield AI

👥201-500
📍San Diego, California, United States
🔗Website

Shield AI Service

Shield AI product / service
Shield AI product / service
Shield AI product / service
Shield AI product / service
Shield AI product / service

How does Shield AI work?

Shield AI builds self-driving technology that enables swarms of unmanned systems to operate on the edge without GPS, communications, or remote pilots.

Company culture

Earn Trust

We live with uncompromising honesty and integrity. We avoid even the appearance of improper behavior.

Candor

We have the courage to address challenges directly and never behind the backs of teammates. We speak up when we disagree. We provide feedback that is clear, direct, and timely.

Visit company profileIconOpenNewWindows

Unlock all Arc benefits!

  • Browse remote jobs in one place
  • Land interviews more quickly
  • Get hands-on recruiter support
PRODUCTS
Arc

The remote career platform for talent

Codementor

Find a mentor to help you in real time

LINKS
About usPricingArc Careers - Hiring Now!Remote Junior JobsRemote jobsCareer Success StoriesTalent Career BlogArc Newsletter
JOBS BY EXPERTISE
Remote Front End Developer JobsRemote Back End Developer JobsRemote Full Stack Developer JobsRemote Mobile Developer JobsRemote Data Scientist JobsRemote Game Developer JobsRemote Data Engineer JobsRemote Programming JobsRemote Design JobsRemote Marketing JobsRemote Product Manager JobsRemote Project Manager JobsRemote Administrative Support Jobs
JOBS BY TECH STACKS
Remote AWS Developer JobsRemote Java Developer JobsRemote Javascript Developer JobsRemote Python Developer JobsRemote React Developer JobsRemote Shopify Developer JobsRemote SQL Developer JobsRemote Unity Developer JobsRemote Wordpress Developer JobsRemote Web Development JobsRemote Motion Graphic JobsRemote SEO JobsRemote AI Jobs
© Copyright 2025 Arc
Cookie PolicyPrivacy PolicyTerms of Service