Senior Machine Learning Engineer

Overview

Role involves designing and scaling novel foundation models in AI.

Ideal candidate has 5+ years of experience in machine learning and proficiency in PyTorch and Ray.

Locations

• Canada, Ontario, Waterloo
• United States, California, San Francisco

Requirements

• Bachelor's degree or equivalent
• 5+ years of experience
• Experience with distributed training pipelines
• Proficiency in Python, Ray Trainer, PyTorch
• Familiarity with cloud platforms

Responsibilities

• Design and implement state-of-the-art models
• Productionize research codes and models
• Shape benchmark design and evaluation frameworks
• Build agentic AI capabilities
• Collaborate with researchers to scale models