Note: The job is a remote job and is open to candidates in USA. The employer is transforming government agencies by digitizing forms and automating workflows with artificial intelligence. The Senior AI/ML Engineer will architect and build production machine learning capabilities, including vector search, retrieval-augmented generation, fine-tuned LLMs, and human feedback loops, while integrating intelligent behaviors into a no-code form builder.
Responsibilities
- Build and tune vector-based retrieval pipelines using OpenAI embeddings and Azure AI Search
- Design prompt strategies and agents to translate parsed PDF data into form component schemas
- Fine-tune LLMs for structured output generation with low-latency performance in mind
- Lead the development of an RLHF loop that incorporates builder UI feedback and audit data
- Help architect systems that blend traditional APIs and probabilistic inference reliably
- Work alongside full-stack and platform engineers to get it all running in production
- Stay plugged into the latest model capabilities, and make smart calls on what to adopt
Skills
- 5+ years in applied ML, including experience with retrieval, embeddings, and prompt engineering
- Strong Python skills and familiarity with production-grade ML pipelines
- Experience designing and tuning RAG workflows with hybrid search
- Familiarity with RLHF and fine-tuning on structured JSON output
- Solid grasp of system-level thinking-how to bring ML into product environments cleanly
- Nice to have: .NET understanding, especially for integration and orchestration layers
Benefits
- 100% remote based in the US
- Help shape the AI transformation of public sector services
- Lead initiatives that ship real impact, not just prototypes
- Greenfield development on a proven, profitable platform
- Comprehensive Health, Vision, Dental coverage for individuals and families
Company Overview
Company H1B Sponsorship