Snorkel AI

The data-centric AI platform powered by programmatic labeling and foundation models

Applied AI Engineer – Federal

AI EngineerMachine Learning EngineerFull TimeRemoteTeam 51-200Since 2019H1B SponsorCompany SiteLinkedIn

Location

District of Columbia + 1 moreAll locations: District of Columbia, Washington

Posted

110 days ago

Salary

$160K - $250K / year

Bachelor Degree3 yrs expEnglishAirflowPandasPythonPy TorchRayScikit LearnSparkType Script

Job Description

• Partner with customers to build and deploy impactful Gen AI and machine learning solutions, from use case scoping and data exploration to model development and deployment. This may involve leveraging Snorkel Flow or designing custom approaches using state-of-the-art tools, with the goal of delivering real business value and informing the evolution of the Snorkel platform. • Develop and implement state of the art AI systems such as retrieval-augmented generation (RAG), fine-tuning pipelines, prompt engineering recipes and agentic workflows. • Create augmented real-world datasets and comprehensive evaluation workflows to ensure model reliability, transparency, and stakeholder trust. A data- and evaluation-first mindset is essential for success in this role. • Forge and manage relationships with our customers’ leadership and stakeholders to ensure successful development and deployment of AI projects with Snorkel Flow. • Collaborate closely with pre-sales Solutions and Product teams to map customer needs to existing capabilities, prioritize roadmap gaps, and guide successful project setup. • Work with other Applied AI Engineers to standardize solutions and contribute to internal tooling and best practices. • Lead stakeholder education on quantitative capabilities, helping them to understand the strengths and weaknesses of different approaches and what problems are best-suited for Snorkel AI. • Serve as the voice of our customers for new AI paradigms, data science workflows, and share customer feedback to product teams. • Conduct one-to-few and one-to-many enablement workshops to transfer knowledge to customers considering or already using Snorkel AI. • As part of our team, you will have the opportunity to work on all aspects of complex National Security problems. • Annual travel up to 25%.

Job Requirements

  • B.S. degree in a quantitative field such as Computer Science, Engineering, Mathematics, Statistics, or comparable degree/experience.
  • 3+ years of customer-facing experience in the design and implementation of AI/ML solutions.
  • Proficiency in Python, including strong grounding in software engineering fundamentals (e.g., modular design, testing, profiling, packaging) and experience with modern Python constructs and libraries for type validation and typed data modeling (e.g., pydantic), building type-safe systems (e.g., mypy), testing (e.g., pytest), packaging and environment configuration (e.g., poetry), API and service frameworks (e.g., FastAPI), serialization and structured data handling (e.g., msgspec), and orchestration tooling relevant to ML deployment (e.g., Ray, Airflow).
  • Expertise across the Applied AI stack, spanning classical ML libraries (e.g., scikit-learn), deep learning frameworks (e.g., PyTorch), foundation-model ecosystems (e.g., Hugging Face Transformers), vector/embedding tooling (e.g., FAISS), data processing frameworks (e.g., pandas, Spark), retrieval/RAG tooling (e.g., Chroma, Weaviate), synthetic dataset curation, evaluation workflows, and LLM orchestration, workflow, agent authoring tools (e.g., LlamaIndex, LangGraph, CrewAI).
  • Experience leading strategic, customer-facing initiatives and collaborating with business stakeholders to ensure ML solutions drive successful business outcomes, with a strong focus on teaching and enablement.
  • Outstanding presentation skills to technical and executive audiences, whether impromptu on a whiteboard or using presentations and demos.
  • Ability to work in a fast-paced environment and balance priorities across multiple projects at once.
  • Experience reviewing and drafting responses to federal Requests for Comment (RFCs), Requests for Information (RFIs), Requests for Proposals (RFPs), etc. is preferred
  • Experience working across the Civilian, DOD and NATSEC agencies, inclusive of experience in Federal Law Enforcement, Healthcare, Financial/Regulatory and Military operations and logistics
  • Candidates must have an active TS/SCI; preferred with a Single Scope Background Investigation (SSBI) with Polygraph.

Benefits

  • Employee stock options

Related Job Pages

More AI Engineer Jobs

Senior AI Engineer, Prior Authorization

Arbiter

Unifying and automating referral workflows so that every patient reaches the right provider at the right time and cost.

AI Engineer110 days ago
Full TimeRemoteTeam 11-50Since 2025

Senior AI Engineer building AI systems for healthcare applications

BigQueryCloudGoogle Cloud PlatformPython
United States
$180K - $240K / year

Senior AI Engineer

Arbiter

Unifying and automating referral workflows so that every patient reaches the right provider at the right time and cost.

AI Engineer110 days ago
Full TimeRemoteTeam 11-50Since 2025

Senior AI Engineer building intelligent healthcare automation systems

BigQueryCloudGoogle Cloud PlatformPython
United States
$180K - $240K / year

Principal AI Software Engineer – Enterprise AI Platform

Natera

We are a global leader in cell-free DNA (cfDNA) testing, dedicated to oncology, women’s health, and organ health.

AI Engineer111 days ago
Full TimeRemoteTeam 1,001-5,000Since 2004H1B Sponsor

Enterprise AI Platform Engineer responsible for building Natera's AI-platform.

AirflowAWSCloudDistributed SystemsDockerKubernetesPython
United States
$174.4K - $218K / year

Senior AI Engineer

EverCommerce

Software that Powers the Service Economy

AI Engineer114 days ago
Full TimeRemoteTeam 1,001-5,000Since 2016H1B Sponsor

Senior AI/ML Engineer III leading technical strategy for AI platform initiatives

Distributed SystemsJavaScriptNode.jsPythonPyTorchTypeScript
United States
$145K - $165K / year