Senior Software Developer, ML Ops

Wealthsimple; United States; Hybrid; USD 180,000 - 230,000; Full time

Benefits offered by this job

Health benefits

Life insurance

Employer-sponsored retirement plan

Vacation days

Wellness days

Unlimited sick days / mental health

Travel flexibility up to 90 days per

Job summary

Wealthsimple’s Machine Learning Platform Team is building a multi-product company with Invest, Cash, Crypto, Trade and Tax products. As a Software Engineer on the ML Platform team, you will help enable data scientists to do their best work by improving ML-ops tooling and supporting new data products.

You will collaborate with area owners, data scientists and other engineers to define architecture, drive scalability and reliability across services, and ship production-grade ML tooling at scale.

Qualifications

• 5+ years of experience implementing production-ready services.
• 3+ years of Python experience.
• 5+ years of software development experience in ML infra, ML tooling or ML engineering.
• Proven experience designing and implementing highly performant scalable systems with good observability.
• Proven experience owning a project end-to-end and tending complex systems for the long run.
• Preferred experience with ML ecosystem (e.g. PyTorch, transformers, Ray, feature stores, elastic search and similar vector databases, Triton, CUDA, DVC, vLLM).

Responsibilities

• Partner with stakeholders across product domains and infrastructure to tackle technical problems at scale, building framework-agnostic services for ML needs.
• Establish roadmap and architecture for Wealthsimple's Machine Learning platform.
• Build highly performant, scalable systems, contributing to ML platform on Kubernetes and Ray.
• Work with autonomy to set the technical direction for scalable projects.

Skills

Production-grade services

Python

ML infrastructure

Observability

End-to-end ownership

ML ecosystem (PyTorch, Transformers, R

Tools

Kubernetes

FastAPI

MLflow

Ray

Bedrock

vLLM

Kafka

Postgres

Redshift

Snowflake

Chronon Feature store

Job description

Build something people love

Wealthsimple is Canada’s leading financial innovator. The company offers a full suite of simple, sophisticated financial products across managed investing, do-it-yourself trading, cryptocurrency, tax filing, spending and saving. Wealthsimple currently serves more than 4 million Canadians and holds over $155 billion in assets under administration. The company was founded in 2014 by a team of financial experts and technology entrepreneurs, and is headquartered in Toronto, Canada.

We're proud of what we've built — and we're just getting started. Read our Culture Manual and learn more about how we work.

Machine Learning Platform Team

The Machine Learning Platform (MLP) team builds the foundation that every ML and GenAI workload at Wealthsimple runs on. We partner with data science and engineering teams across the company to take models from experimentation to production reliably and at scale. Our responsibilities are to:

• Build standardized, framework-agnostic services for model training, efficient inference, evaluation and monitoring — across both traditional ML and GenAI


• Provide centralized, governed access to internal and external LLMs


• Set the technical direction for how ML and LLM models are trained, shipped and operated across the company

The Role

We are building a multi-product company with a growing offering including Invest, Cash, Crypto, Trade and Tax products.

As a Software Engineer on the Machine Learning Platform team, you will play a pivotal role in our mission to build a self-serve ML Platform to enable data scientists to do their best work. As such, you will work closely with the area owners, data scientists and other software engineers to improve our ML-ops tooling and support the launch of new data products.

In this role, you will have the opportunity to:

• Partner with stakeholders across different product domains, infrastructure to tackle technical problems at scale, which includes building framework agnostic services to support all of Wealthsimple's machine learning needs.


• Establish roadmap and architecture for Wealthsimple's Machine Learning platform


• Build highly performant scalable systems, contributing to our ML platform on Kubernetes and Ray


• Work with a large amount of autonomy to set the technical direction for scalable projects.

We are looking for people who have:

• 5+ years of experience implementing production-ready services


• 3+ years of experience with python


• 5+ years of software development experience in ML infra, ML tooling or ML engineering


• Proven experience designing and implementing highly performant scalable systems with good observability


• Proven experience owning a project end-to-end and owning and tending complex systems for the long run


• Preferred experience with ML ecosystem (e.g. pytorch, transformers, Ray, feature stores, elastic search and similar vector databases, triton, cuda, DVC, vLLM)

Our stack includes:

• Kubernetes to orchestrate our services


• FastAPI for RESTful APIs


• MLflow for experiment tracking


• Kubernetes, Ray, Bedrock, vLLM for ML training and inference


• Chronon Feature store


• Kafka, postgres, redshift and snowflake for streaming and data warehousing

Why Wealthsimple?

Top-tier health benefits and life insurance

Long-term group savings with employer match, through Wealthsimple for Business

20 vacation days, 4 wellness days, and unlimited sick and mental health days per year

90 days away: work outside Canada for up to 90 days per year

Employee resource groups, including Rainbow (2SLGBTQ), Women of WS, and Black at WS

We are a hybrid team with over 1,500 employees across North America. The people are one of the best parts of working here: you'll collaborate with incredibly talented, curious, and driven teammates who are deeply committed to doing great work.

ICYMI

Technology & Innovation at Wealthsimple: We move quickly and build thoughtfully. That means we're always looking for better ways to work — whether that's new tools, AI, or rethinking how we approach a problem. We don't expect you to have all the answers, but we do expect curiosity and a willingness to evolve alongside the products we're building.

Inclusion Statement: We're building products for a diverse world, and we need a diverse team to do it well. We strongly encourage applications from everyone, regardless of race, religion, colour, national origin, gender, sexual orientation, age, marital status, or disability status.

Accessibility Statement: We're committed to an accessible hiring experience. If you need any accommodations throughout the interview process, please let us know — we'll work with you to make sure you have what you need. We also welcome any feedback on how we can better accommodate candidates with accessibility needs.

AI in Hiring: We may use artificial intelligence (AI) tools to support parts of our hiring process, such as reviewing applications, analyzing resumes, or assessing responses. These tools assist our team but don't replace human judgment – all final hiring decisions are made by people. If you have questions about how your data is used, reach out to us.