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 Infrastructure Team
The Machine Learning Infrastructure team builds the core foundation powering AI and GenAI initiatives across Wealthsimple. As our AI footprint expands, our priority is building scalable, cost-effective infrastructure that enables high-performance inference using open-source models alongside vendor solutions. Our responsibilities are to:
Build and maintain infrastructure to host and serve open-source LLMs reliably on cloud/GPU systems.
Implement intelligent model-routing frameworks to optimize performance, cost, and latency.
Develop evaluation and benchmarking tooling (Evals) to test, validate, and compare model performance across use cases.
Set the technical direction for operating scalable LLM infrastructure company-wide.
The Role
As a Senior Software Engineer on the ML Infrastructure team, you will design and implement the core backend and infrastructure powering our AI systems. You will work closely with data scientists, platform engineers, and engineering leaders to scale our GenAI footprint efficiently and securely.
In this role, you will have the opportunity to:
Build model-routing architecture (e.g., LiteLLM integration) to dynamically direct requests across internal and external models.
Provision and manage high-performance GPU infrastructure for serving open-source LLMs.
Build evaluation frameworks (Evals) that enable engineers and data scientists to benchmark model quality against production requirements.
Drive key cost-optimization and infrastructure efficiency initiatives without sacrificing system accuracy or reliability.
We are looking for people who have:
6+ years of software engineering experience building production-ready distributed systems or data platform infrastructure.
Strong proficiency in Python and infrastructure tooling like Kubernetes and Terraform.
Proven experience designing and operating highly performant, observable backend systems.
Practical experience or deep familiarity with the modern AI ecosystem (e.g., serving open-source models, vLLM, Ray, model routing, or evaluation frameworks).
Track record of owning technical projects end-to-end with high autonomy.
Our stack includes:
Container & Infra: Kubernetes, Terraform, AWS GPU infrastructure
Languages & APIs: Python, FastAPI
LLM Tooling & Serving: LiteLLM, vLLM, Ray, Bedrock
Data & Streaming: Kafka, Postgres, Redshift, Snowflake
🌸 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.
ICYMITechnology & 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.
Skills Required
- 5+ years implementing production-ready services
- 3+ years of experience with Python
- 5+ years software development experience in ML infrastructure, ML tooling, or ML engineering
- Proven experience designing and implementing highly performant, scalable systems with good observability
- Proven experience owning projects end-to-end and tending complex systems long-term
- Experience with Kubernetes
- Experience with Ray
- Preferred experience with ML ecosystem: PyTorch, Transformers, Ray, feature stores, Elasticsearch or similar vector DBs, Triton, CUDA, DVC, vLLM
- Familiarity with stack tools (FastAPI, MLflow, Bedrock, Chronon Feature Store, Kafka, Postgres, Redshift, Snowflake)
Wealthsimple Compensation & Benefits Highlights
The following summarizes recurring compensation and benefits themes identified from responses generated by popular LLMs to common candidate questions about Wealthsimple and has not been reviewed or approved by Wealthsimple.
-
Leave & Time Off Breadth — Time off programs include generous vacation, unlimited sick and mental health days, and a “90 Days Away” option to work internationally. These policies signal broad flexibility and ample opportunities for rest and renewal.
-
Parental & Family Support — Paid parental leave is topped up to full salary for extended periods for both primary and secondary caregivers. Structured reboarding and gradual return options reinforce support for growing families.
-
Wellbeing & Lifestyle Benefits — Offerings include a substantial mental health budget, Headspace access, wellness and home office stipends, and a remote‑friendly setup. These benefits emphasize whole‑person wellbeing beyond core pay and insurance.
Wealthsimple Insights
What We Do
All of your investing, made simple. wsim.co/legal
.png){kind=logo}
