Applied AI Engineer - Federal

Snorkel AI is on a mission to democratize AI by building a leading AI data development platform. As an Applied AI Engineer, you will research and utilize advanced Gen AI and machine learning techniques to deliver tailored solutions for customers, while also contributing to the evolution of Snorkel's Applied AI tooling.

Responsibilities:

• 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%

Requirements:

• 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
• Candidates must have an active TS (Top Secret) Clearance
• 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
• Experience reviewing and drafting responses to federal Requests for Comment (RFCs), Requests for Information (RFIs), Requests for Proposals (RFPs), etc
• Experience working across the Civilian, DOD and NATSEC agencies, inclusive of experience in Federal Law Enforcement, Healthcare, Financial/Regulatory and Military operations and logistics