Senior AI Engineer, AI Platform
Affinity.co
full-time
Posted on:
Location Type: Hybrid
Location: San Francisco • California • United States
Visit company websiteExplore more
Salary
💰 $160,000 - $235,000 per year
Job Level
About the role
- Architect, prototype, and deploy RAG pipelines, combining vector search, hybrid retrieval, reranking and contextual compression techniques.
- Contribute to design and orchestration of multi-agent LLM systems using community frameworks and custom orchestration layers.
- Work on a variety of information extraction, information storage and information retrieval problems for both structured and unstructured data.
- Partner with cross-functional (product, infra, data engineering, and software engineering) to build robust, high-scale systems that underlie all of our data processing and ML Operations.
Requirements
- 5+ years of experience in software engineering and/or Machine Learning experience in applying machine learning in production.
- Hands on experience with LLM applications in production including prompt engineering and utilizing frameworks for online and offline evaluation
- Experience with LLM assisted search, such as query understanding and augmentation, text2sql, and entity extraction
- Experience with vector or graph databases
- Experience with document chunking, embedding models, and context window optimization
- Familiarity with metadata-based retrieval and re-ranking strategies
- Hands on experiences with model evaluation metrics (e.g. perplexity, hallucination rate, factual consistency)
- Familiarity with data security, versioning, and MLOps principles.
Benefits
- We cover your medical, dental, and vision insurance premiums with comprehensive PPO, HDHP and HMO options (in CA), and offer flexible personal & sick days to support your well-being.
- We offer a 401(k) plan to help you plan for your future.
- We provide an annual education budget and a comprehensive L&D program.
- We reimburse monthly for things like home internet, meals, and wellness memberships/equipment to support your overall health and happiness.
- Virtual team-building activities and socials to keep our team connected, because building strong relationships is key to success.
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard Skills & Tools
RAG pipelinesvector searchhybrid retrievalreranking techniquescontextual compressioninformation extractioninformation storageinformation retrievalLLM applicationsprompt engineering
Soft Skills
cross-functional collaborationdesignorchestration