Machine Learning Engineer, AI/ML

We're looking for a Machine Learning Engineer to join our team in Toronto, focusing on building and optimizing state-of-the-art LLM-powered agents that can reason, plan and automate workflows for users. You'll be joining us at an exciting time as we reinvent our insight generation systems, making this an excellent opportunity for someone with strong Backend and ML fundamentals who wants to dive deep into practical LLM applications.

Requirements

• 5+ years of software engineering experience
• Experience with information retrieval systems, search relevance, and ranking models
• Expertise in Python, with experience in frameworks such as PyTorch, TensorFlow, or JAX
• Familiarity with LLMs, prompt engineering, and retrieval-augmented generation pipelines
• Understanding of evaluation methods for search systems, including offline metrics and user-facing evaluation
• Experience working with vector database infrastructure and traditional search engines
• Understanding of data pipelines, preprocessing, and large-scale data handling
• Ability to work independently and collaboratively in a fast-paced environment, balancing research and production needs
• Develop and implement CI/CD pipelines. Automate the deployment and monitoring of ML models
• Knowledge of query understanding, document summarization and other 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 with cloud infrastructure for scalable ML workflows

Benefits

• Competitive base salary
• Benefits. Extended health & dental benefits that kick in Day 1
• Options. Opportunity to participate in our Employee Stock Option Plan
• Time off. Take what you need. Just ensure the required work gets done and clear it with your team in advance. The average Klue team member takes 2-4 weeks of PTO per year
• Direct access to our leadership team, including our CEO