Senior AI Engineer

Affinity stitches together billions of data points from massive datasets to create a powerful, accurate representation of the world's professional relationship graph. Based on this data, we offer our users the insights and visibility they need to nurture and tap into the opportunities in their team's network.

• Build RAG systems: Architect, prototype, and deploy RAG pipelines, combining vector search, hybrid retrieval, reranking and contextual compression techniques.
• Build LLM powered agent systems: Contribute to design and orchestration of multi-agent LLM systems using community frameworks and custom orchestration layers.
• Solve complex problems: Work on a variety of information extraction, information storage and information retrieval problems for both structured and unstructured data.
• Collaborate cross-functionally: Partner with product, infra, data engineering, and software engineering to build robust, high‑scale systems that underlie all of our data processing and ML operations.

Don't meet every single requirement? Studies have shown that women and people of color are less likely to apply to jobs unless they meet every qualification. At Affinity, we are dedicated to building a diverse, inclusive, and authentic workplace, so if you're excited about this role, but your past experience doesn't perfectly align with the qualifications above, we encourage you to apply anyways.

You may be just the right candidate for this or other roles.

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

• Experience with enterprise AI applications with strict compliance, audit, or legal requirements.
• Experience with dataset engineering, including data curation, augmentation, and synthesis, to assist ML model improvements.
• Experience with multi‑modal search.
• Experience with graph‑based recommendation systems, such as graph NN.
• Experience with developing AI applications powered by agent‑based systems.
• Experience with packaging, CI/CD and pipeline automation.

Our ML pipeline manages multiple Python services that support various AI features, including utilizing OCR to extract information from unstructured data, serving embedding models to vectorize chunks, and ranking a list of recommendations based on relevance and user preference.

• Health Benefits: We cover both you and your dependents' extended health benefit premiums and offer flexible personal & sick days to support your well‑being.
• Retirement Planning: We offer an RRSP plan to help you plan for your future.
• Learning & Development: We provide an annual education budget and a comprehensive L&D program.
• Wellness Support: We reimburse monthly for things like home internet, meals, and wellness memberships/equipment to support your overall health and happiness.
• Team Connection: Virtual team‑building activities and socials to keep our team connected, because building strong relationships is key to success.

Salary range: $160,000 to $220,000 CAD. Within the range, individual pay depends on various factors including geographical location and review of experience, knowledge, skills, and abilities of the applicant.

At Affinity, we may use artificial intelligence (AI) tools as part of our recruitment process to help screen and evaluate candidate applications. While AI assists our hiring team in managing applications efficiently, it never replaces decisions made by real people. We are committed to fair and transparent hiring practices, and our AI tools are regularly monitored to ensure they support—not replace—human judgment.

We use E‑Verify to confirm the employment eligibility of all newly hired employees. To learn more about E‑Verify, including your rights and responsibilities, please visit (Use the "Apply for this Job" box below)..