Senior ML Software Developer

Primate Labs, a small (but mighty!) software company located in Midtown Toronto, is looking to hire an enthusiastic Senior Machine Learning Software Developer. The developer will contribute to Geekbench AI, Primate Labs’ cross-platform AI benchmark, by investigating, designing, and prototyping machine learning workloads that measure how well different hardware platforms execute neural networks.

This position is an excellent opportunity for developers interested in machine learning, computer performance, and cross-platform development.

Responsibilities

As a Senior Machine Learning Software Developer, your primary responsibility will be to investigate, design, and prototype machine learning models and benchmark tests. The tests draw from a variety of fields including traditional machine learning tasks (e.g., object detection, image segmentation, text classification) and generative AI tasks (e.g., text generation using SLMs and LLMs). The tests use either in-house or pre-trained models.

You will also be responsible for communicating designs and experimental results from the workloads both internally to project stakeholders and externally to analysts, journalists, and representatives and researchers from hardware companies.

Qualifications

The ideal candidate has the following background and skills:

• Experience with Python.
• Experience evaluating and training machine learning models.
• Experience with at least one machine learning toolkit (e.g., TensorFlow, PyTorch).
• Strong written and oral communication skills.
• Able to work independently with minimal supervision.
• Able to conduct research and problem solve independently
• An undergraduate degree in a technical field (e.g., Computer Engineering, Computer Science, Software Engineering).
• At least three years of industry machine learning experience.

The following skills are nice to have but are not necessary:

• Experience with C++.
• Experience with at least one inference framework (e.g., Core ML, ONNX, OpenVINO).
• Experience quantizing machine learning models.
• Experience evaluating generative AI models (e.g., SLMs, LLMs).
• A graduate degree in a technical field.

This is a hybrid job in Primate Labs' Toronto office.

Primate Labs offers a fun and exciting working environment with minimal bureaucracy. Compensation includes a competitive salary, generous vacation time, and medical and dental benefits.