Senior ML Systems Engineer - Scalable AI Deployments

The Institute for Performance and Learning; Montreal (administrative region); Hybrid; CAD 120,000 - 160,000; Full time

Job summary

DRW in Montreal is seeking a Machine Learning Specialist to build critical systems for ML models and drive impactful projects. You will work with cutting-edge tech and colleagues across ML, data, and software to advance production-grade solutions.

Responsibilities include scalable model serving, training optimization, feature stores, and retraining pipelines. Strong ML foundations and GPU programming experience are essential.

Qualifications

• Bachelor's or Master's degree in computer science, software engineering or equivalent.
• 5+ years of experience building software for machine learning systems.
• Strong understanding of machine learning and deep learning concepts.
• Excellent programming and scripting skills.
• Experience with building, validating, deploying, monitoring, and updating production models.
• Experience with infrastructure and tooling for MLOps.
• Strong problem-solving and analytical skills.
• Excellent communication and collaboration skills.
• Experience with machine learning frameworks such as TensorFlow, PyTorch, TensorRT, ONNX.
• Experience with GPU programming using CUDA, OpenCL or similar libraries.
• Experience optimizing inference performance (quantization, profiling, kernel/runtime tuning, serving engines).
• Experience with one or more of the following is an asset: distributed systems, multi-threaded programming, hardware/software interaction

Responsibilities

• Design and implement scalable and performant systems to serve machine learning models
• Optimize training and inference performance of deep learning models
• Create and maintain feature pipelines, feature stores, model stores
• Monitor production models and setup retraining infrastructure to improve performance
• Optimize models for deployment on different hardware configurations
• Build frameworks to promote scalable and reproducible research
• Drive projects and lead new initiatives
• Facilitate discussions on code implementation, testing, software architecture best practices
• Actively engage with others on the team to resolve challenging design and coding issues
• Collaborate with stakeholders from different departments

Job description

DRW in Montreal is seeking a Machine Learning Specialist to build critical systems for ML models and drive impactful projects. You will work with cutting-edge tech and colleagues across ML, data, and software to advance production-grade solutions.

Responsibilities include scalable model serving, training optimization, feature stores, and retraining pipelines. Strong ML foundations and GPU programming experience are essential.