Senior Machine Learning Engineer, RAG
Klue is a VC-backed, capital-efficient growing SaaS company creating the category of competitive enablement, helping companies understand their market and outmaneuver their competition.
$144,000 - $160,000
Machine Learning
Senior Software Engineer
Hybrid
2+ years of experience
AI · Enterprise SaaS
Description For Senior Machine Learning Engineer, RAG
Klue is seeking a Senior Machine Learning Engineer to join their ML team in Toronto, focusing on building and optimizing state-of-the-art RAG (Retrieval Augmented Generation) systems. This role offers an exciting opportunity to reinvent RAG systems, ideal for someone with strong ML and IR fundamentals who wants to dive deep into practical LLM applications.
Key Responsibilities:
- Optimize RAG systems with scientific rigor and reproducible results
- Measure and improve retrieval systems using comprehensive evaluation metrics
- Develop optimal chunking and enrichment strategies for diverse data sources
- Work on prompt engineering to effectively utilize retrieved context
- Train and fine-tune smaller, more efficient models
- Deploy and monitor models in production, optimize latency, and implement metrics
- Apply latest research breakthroughs to improve systems
- Collaborate with backend engineers to build scalable, production-ready systems
Required Experience:
- Masters or PhD in Machine Learning, NLP, or related field
- 2+ years building and optimizing retrieval systems
- 2+ years training/fine-tuning transformer models
- Strong foundation in evaluating RAG systems
- Deep understanding of retrieval metrics and their trade-offs
- Expertise in automated LLM evaluation and prompt engineering
- Experience deploying models to production and monitoring system health
Technologies Used:
- LLM platforms: OpenAI, Anthropic, open-source models
- ML frameworks: PyTorch, Transformers, spaCy
- Search/Vector DBs: Elasticsearch, Pinecone, PostgreSQL
- MLOps tools: Weights & Biases, MLflow, Langfuse
- Infrastructure: Docker, Kubernetes, GCP
- Development: Python, Git, CI/CD
Klue offers a hybrid work environment with main Canadian hubs in Vancouver and Toronto. The role is ideally located in Toronto, with at least 2 days per week in the office. Join a company recognized as one of Canada's Most Admired Corporate Cultures, a Deloitte Technology Fast 50 & Fast 500 winner, and recipient of Startup of the Year and Tech Culture of the Year awards.
Last updated 2 months ago
Responsibilities For Senior Machine Learning Engineer, RAG
- Optimize RAG systems with scientific rigor and reproducible results
- Measure and improve retrieval systems across the spectrum from BM25 to semantic search
- Develop optimal chunking and enrichment strategies for diverse data sources
- Work on prompt engineering to effectively utilize the retrieved context
- Train and fine-tune smaller, more efficient models
- Deploy and monitor models in production, optimize latency, and implement comprehensive metrics
- Apply deep understanding of latest breakthroughs to improve systems
- Collaborate with backend engineers to build scalable, production-ready systems
Requirements For Senior Machine Learning Engineer, RAG
Python
PostgreSQL
- Masters or PhD in Machine Learning, NLP, or related field
- 2+ years building and optimizing retrieval systems
- 2+ years training/fine-tuning transformer models
- Strong foundation in evaluating RAG systems
- Deep understanding of retrieval metrics and their trade-offs
- Strong grasp of embedding models, semantic similarity techniques, and clustering similar content
- Knowledge of query augmentation and content enrichment strategies
- Expertise in automated LLM evaluation, including LLM-as-judge methodologies
- Skilled at prompt engineering - including zero-shot, few-shot, and chain-of-thought
- Experience deploying models to production and monitoring the health of the system and the predictions
- Knowledge of ML infrastructure, model serving, and observability best practices
- Proven ability to balance scientific rigor with driving business impact
- Track record of staying current with ML research and breakthrough papers
Benefits For Senior Machine Learning Engineer, RAG
Equity
- Equity
Jobs Related To Klue Senior Machine Learning Engineer, RAG
Explore Jobs By LevelEntry-Level Software Engineer JobsMid-Level Software Engineer JobsSenior Software Engineer JobsStaff Software Engineer Jobs
Explore TrendingLayoffsPerformance Improvement PlanSystem DesignInterpersonal CommunicationTech Lead
