Latest AI JobsBrowse Jobs
wealthsimple logo

Senior Software Developer, ML Platform & Infrastructure

wealthsimpleToronto HeadquartersData & Engineering
Apply on company site
Salary
CA$152K – CA$189K
Job type
Full-time
Experience
Senior · 7+ yrs
Posted
2 Oct 2026
Status
Open · checked 7d ago

About this role

Based at the Toronto Headquarters, the Senior Software Developer, ML Platform & Infrastructure role at Wealthsimple suits an experienced software engineer specializing in MLOps and infrastructure who wants to transition their expertise into large language model serving and GenAI platform scaling. This permanent full-time position places the candidate within a major Canadian financial innovator, directing the architecture that powers modern artificial intelligence initiatives.

Role Overview

The Machine Learning Infrastructure and Platform team seeks a senior engineer to take end-to-end ownership of the technical direction for operating enterprise-grade large language model systems. This position bridges traditional MLOps with modern generative AI stack requirements. The selected candidate will build self-serve systems, manage GPU infrastructure, and optimize inference performance for internal engineering and data science teams.

Key Responsibilities

Transition traditional machine learning lifecycles and serving patterns into state-of-the-art inference engines and GPU orchestration systems.
Design low-latency routing frameworks to dynamically direct requests across managed cloud providers and self-hosted open-source models.
Architect and manage high-performance GPU serving environments on Kubernetes using specialized serving engines.
Build automated evaluation and observability frameworks to validate model quality, latency, and drift against production requirements.
Partner with product engineering and data science teams to deliver framework-agnostic platform tooling that abstracts infrastructure complexity.
Optimize price-performance ratios across inference systems through improved capacity utilization, batching, and routing.

Requirements

Seven or more years of software engineering experience focusing on machine learning infrastructure, MLOps, machine learning tooling, or data platform engineering.
Proven track record of building and operating self-serve machine learning platforms, model registry workflows, experiment tracking, or production serving infrastructure.
Advanced proficiency in Python, container orchestration via Kubernetes, infrastructure-as-code using Terraform, and cloud provider ecosystems like AWS.
Strong desire to specialize in large language model serving, including vLLM, model routing, vector databases, and GPU memory optimization.
Experience designing highly available, observable microservices handling real-time and low-latency requests.
Proven capability to lead architectural roadmaps, guide multi-functional projects with high autonomy, and maintain complex platform systems.

Nice to Have

Direct experience serving open-source large language models in production environments.
Hands-on work with CUDA, GPU partitioning, or distributed inference frameworks.
Familiarity with vector search and retrieval engines like Elasticsearch, Qdrant, or Pinecone.

Skills and Tools

Python, FastAPI, PyTorch
Kubernetes, Terraform, AWS GPU Infrastructure
vLLM, Ray, Triton, LiteLLM, AWS Bedrock, MLflow, SageMaker
Kafka, Postgres, Redshift, Snowflake

Salary and Benefits

The salary range for this position is CA$152,000 to CA$189,000 alongside equity offerings. Wealthsimple provides top-tier health benefits, life insurance, and long-term group savings with an employer match. Employees receive 20 vacation days, 4 wellness days, and access to employee resource groups.

About the Company

Wealthsimple operates as a leading financial innovator in Canada, offering managed investing, self-directed trading, cryptocurrency, tax filing, and spending products. Founded in 2014, the organization serves over 4 million Canadians and holds more than $155 billion in assets under administration from its headquarters in Toronto, Canada.

How to Apply

Interested candidates should apply directly on the official company careers page.

Skills and tech stack

PythonFastAPIPyTorchKubernetesTerraformAWSvLLMRayTritonLiteLLMAWS BedrockMLflowSageMakerKafkaPostgresRedshiftSnowflakeLLMLLMsMLOpsMachine LearningCUDAPrompt Engineering

Questions candidates ask

Where is this job located?

This role is based at the Wealthsimple headquarters in Toronto, Canada.

What is the salary range for this position?

The salary ranges from CA$152,000 to CA$189,000, and the position also offers equity.

How many years of experience are required?

Candidates must have 7 or more years of software engineering experience in machine learning infrastructure, MLOps, machine learning tooling, or data platform engineering.

What technologies and tools are part of the stack?

The stack includes Python, FastAPI, PyTorch, Kubernetes, Terraform, AWS, vLLM, Ray, Triton, LiteLLM, AWS Bedrock, MLflow, SageMaker, Kafka, Postgres, Redshift, and Snowflake.

What benefits are included?

Benefits include top-tier health benefits, life insurance, long-term group savings with an employer match, 20 vacation days, and 4 wellness days.

Ready to apply?

The application is on the official wealthsimple careers page.

Apply on company site