Senior Machine Learning Engineer – AIOS Platform

About the role

Zafin is an AI platform company helping regulated institutions modernize critical work. As a Senior Machine Learning Engineer on the AIOS product team, you will build the ML and AI systems that power the AIOS Knowledge Fabric, enabling agents to reason over enterprise knowledge with accuracy, traceability, and control.

Key responsibilities

• Design, build, test, and maintain ML and AI capabilities for AIOS, focusing on knowledge retrieval, model integration, and production reliability.
• Develop systems for embeddings, vector search, ranking, retrieval‑augmented generation, knowledge ingestion, and quality improvement.
• Build and improve model evaluation workflows, test sets, scoring approaches, quality metrics, and monitoring for AI and LLM‑based systems.
• Partner with AI engineers, data engineers, product managers, architects, and domain experts to translate product needs into ML system designs and working software.
• Write production‑quality Python code, including services, pipelines, automation, tests, and integration points with product systems.
• Improve ML system performance, latency, cost, reliability, and observability in production and pre‑production environments.
• Support responsible AI practices by improving traceability, explainability, governance, auditability, privacy, and human review workflows.

Required profile

• 5+ years of software engineering, machine learning engineering, data science engineering, or applied ML experience building production systems.
• 2+ years of hands‑on experience with production ML, LLM applications, retrieval‑augmented generation, NLP, search, recommendation, or AI‑powered automation.
• Strong programming skills in Python and experience building maintainable, tested, production‑quality code.
• Practical experience with ML frameworks, LLM APIs or open‑source models, embeddings, vector databases, retrieval systems, or model‑serving patterns.
• Bachelor’s degree in Computer Science, Engineering, Machine Learning, Statistics, Mathematics, or related field, or equivalent practical experience.

Required skills

• Python programming
• Machine learning frameworks (e.g., TensorFlow, PyTorch)
• LLM APIs or open‑source models
• Embeddings and vector databases
• Retrieval‑augmented generation and semantic search
• Model serving and MLOps pipelines
• Data pipeline development and feature preparation
• API design, database integration, testing, and version control

What we offer

• Fully company‑paid health benefits and extensive paramedical coverage
• Annual bonus opportunities and RRSP matching
• Generous vacation, holidays and paid volunteering days
• Wellness resources and employer‑paid insurance coverage