Staff Machine Learning Engineer, AI Generation Engine

This is a fully remote position, open to applicants in Canada.

📋 Description

• Design, build, and oversee robust data pipelines for the training, validation, and ongoing retraining of Large Quantitative Models (LQMs) and agentic frameworks.

• Create, implement, and thoroughly evaluate innovative ML models and algorithms, establishing suitable metrics to guarantee that model performance aligns with overarching product goals.

• Spearhead the process of cleaning, transforming, and engineering features from complex, large-scale datasets to enhance LQM performance and predictive accuracy.

• Perform in-depth analyses of model behavior, performance, and failure modes, fine-tuning hyper-parameters and optimizing model architecture for efficiency, speed, and accuracy in a production setting.

• Work collaboratively with AI researchers, product managers, and software engineers to convert high-level business objectives into actionable ML development and deployment plans.

• Advocate for and uphold exceptional engineering standards regarding code quality, system efficiency, and security within a prototyping environment.

• Lead technical execution with significant autonomy, making vital design and implementation choices independently.

⛳️ Requirements

• Bachelor’s degree in Software Engineering, Computer Science, or a related field.

• Over 8 years of postgraduate experience in software development.

• Proven experience in developing highly available, performant, and scalable ML systems, including large-scale data processing pipelines.

• Strong proficiency in Python, encompassing the ML stack: PyTorch, TensorFlow, JAX, NumPy, and Pandas.

• Extensive, successful track record of managing the complete ML lifecycle, from initial data exploration and hypothesis testing to architecture, model training, evaluation, and production deployment.

• In-depth knowledge of MLOps and software best practices, including CI/CD for ML, experiment tracking (e.g., Weights & Biases, MLflow), automated testing, and version control for both code and datasets.

🏝️ Benefits

• Comprehensive medical, dental, and vision coverage for employees and their dependents, with generous employer premium contributions.

• Retirement savings with company matching.

• Paid parental leave.

• Inclusive family-building benefits.

• Flexible paid time off.

• Company-wide seasonal breaks.

• Support for flexible work arrangements that promote sustainable performance.

• Opportunities for continuous learning and growth through on-the-job development, cross-functional collaboration, and access to internal learning and development programs.