Senior Software Developer, ML Platform & Infrastructure

Wealthsimple
Toronto Headquarters, Canada

Who this role is best for

Geared toward candidates with extensive MLOps experience comfortable with LLM inference and GPU orchestration, working in a hybrid setup at Wealthsimple's Toronto headquarters.

Best fit for

  • Candidates with 7+ years in ML Infrastructure or MLOps who are eager to specialize in LLM serving.
    — “7+ years of software engineering experience in ML Infrastructure, MLOps, ML Tooling, or Data Platform engineering.”
  • Individuals with a background in building self-serve ML platforms and production serving infrastructure.
    — “Proven track record building and operating self-serve ML platforms, model registry workflows, experiment tracking, or production serving infrastructure.”
  • Professionals skilled in Kubernetes, Terraform, and cloud environments like AWS, with a focus on backend performance.
    — “Strong platform fundamentals: Advanced proficiency in Python, container orchestration via Kubernetes, infrastructure-as-code (Terraform), and cloud provider ecosystem (AWS).”

Things to consider

  • The role requires a strong appetite to specialize in LLM serving, not just general ML experience.
    — “Strong appetite to specialize in LLM serving: A genuine desire to leverage your existing MLOps skillset to tackle LLM-specific challenges”
  • The position demands end-to-end ownership of technical direction, not just contributing to existing systems.
    — “End-to-end technical ownership: Proven capability to lead architectural roadmaps, guide multi-functional projects with high autonomy”

How to stand out

  • Highlight experience with vLLM, Ray, and Triton in your resume and interview responses.
    — “Provision & scale GPU infrastructure: Architect and manage high-performance GPU serving environments on Kubernetes using engines like vLLM, Ray, and Triton.”
  • Emphasize your history of building automated evaluation and observability frameworks for ML systems.
    — “Develop evaluation & benchmarking tooling: Build automated Evals and observability frameworks to empower engineers and data scientists”
  • Demonstrate your ability to abstract infrastructure complexity for cross-functional teams.
    — “Empower self-serve ML across Wealthsimple: Partner with product engineering and data science teams to build framework-agnostic platform tooling”
Pace · Fast PacedCollaboration · HighAutonomy · MediumDecision Impact · Team

Derived from job-description analysis by Serendipath's career intelligence engine.

What success looks like

  • transition traditional ML lifecycle and serving patterns into state-of-the-art LLM inference engines
  • design low-latency routing frameworks
  • develop evaluation & benchmarking tooling
Typical background
7+ years of software engineering experience in ML Infrastructure, MLOps, ML Tooling, or Data Platform engineering

Skills & requirements

Required

MlopsLLM Inference EnginesGPU Cluster ManagementModel-routing ArchitectureEvaluation & Benchmarking ToolingSelf-serve ML Platforms

Preferred

VllmRayTriton

Stack & domain

MlopsML PlatformModel OrchestrationServing InfrastructureGPU Cluster ManagementIntelligent Model RoutingAutomated EvalsObservability FrameworksKubernetesVllmRayTritonTurning Mlops Expertise To GenaiBuilding Model-routing ArchitectureProvisioning & Scaling GPU InfrastructureDeveloping Evaluation & Benchmarking ToolingEmpowering Self-serve MLDriving Cost & Performance OptimizationSoftware DevelopmentAIGenai

About the role

Original posting from Wealthsimple via Ashby

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 https://www.wealthsimple.com/en-ca/culture and learn more about how we work https://www.wealthsimple.com/en-ca/careers.

ML PLATFORM & INFRASTRUCTURE TEAM

The Machine Learning Infrastructure & Platform team builds the foundational architecture powering AI and GenAI initiatives across Wealthsimple. We sit at the intersection of production MLOps and cutting-edge GenAI enablement.

As our AI footprint expands rapidly, our priority is evolving our robust MLOps foundations into a scalable, high-performance LLM serving and routing platform. We build self-serve systems that allow Data Scientists and Engineers to host open-source LLMs reliably, optimize inference latencies, manage GPU infrastructure, and benchmark model performance safely in production.

THE ROLE

We are looking for an experienced MLOps or ML Platform Engineer who is excited to pivot their deep background in model orchestration, serving, and platform tooling toward solving the unique challenges of LLM inference and infrastructure.

In this role, you will bridge the gap between traditional MLOps (model lifecycles, pipeline orchestration, serving infrastructure) and modern GenAI stack requirements (vLLM, GPU cluster management, intelligent model routing, and automated Evals). You will take end-to-end ownership of setting the technical direction for operating enterprise-grade LLM systems company-wide.

IN THIS ROLE, YOU WILL HAVE THE OPPORTUNITY TO:

  • Turn MLOps expertise to GenAI: Transition traditional ML lifecycle and serving patterns into state-of-the-art LLM inference engines and GPU orchestration systems.
  • Build model-routing architecture: Design low-latency routing frameworks (e.g., LiteLLM integration) to dynamically direct requests across managed cloud providers (AWS Bedrock) and self-hosted open-source models.
  • Provision & scale GPU infrastructure: Architect and manage high-performance GPU serving environments on Kubernetes using engines like vLLM, Ray, and Triton.
  • Develop evaluation & benchmarking tooling: Build automated Evals and observability frameworks to empower engineers and data scientists to validate model quality, latency, and drift against production requirements.
  • Empower self-serve ML across Wealthsimple: Partner with product engineering and data science teams to build framework-agnostic platform tooling that abstracts infrastructure complexity.
  • Drive cost & performance optimization: Improve price-performance across self-hosted and managed inference by optimizing capacity, utilization, batching, routing, and model selection while meeting quality and reliability objectives

WE ARE LOOKING FOR PEOPLE WHO HAVE:

  • 7+ years of software engineering experience in ML Infrastructure, MLOps, ML Tooling, or Data Platform engineering.
  • Deep experience in MLOps/ML Platform practices: Proven track record building and operating self-serve ML platforms, model registry workflows, experiment tracking, or production serving infrastructure (Kubeflow, MLflow, Ray, Triton, SageMaker).
  • Strong platform fundamentals: Advanced proficiency in Python, container orchestration via Kubernetes, infrastructure-as-code (Terraform), and cloud provider ecosystem (AWS).
  • Strong appetite to specialize in LLM serving: A genuine desire to leverage your existing MLOps skillset to tackle LLM-specific challenges (vLLM, model routing, prompt engineering tooling, vector databases, GPU memory optimization, or LLM evaluation frameworks).
  • Backend performance & observability focus: Experience designing highly available, observable microservices (e.g., FastAPI) handling real-time, low-latency requests.
  • End-to-end technical ownership: Proven capability to lead architectural roadmaps, guide multi-functional projects with high autonomy, and maintain complex platform systems for the long run.

NICE-TO-HAVES (OR AREAS YOU WILL LEARN ON THE JOB):

  • Direct experience serving open-source Large Language Models in production (vLLM, SGLang, TensorRT-LLM, Dynamo).
  • Hands-on work with CUDA, GPU partitioning, or distributed inference frameworks (Ray Serve).
  • Familiarity with vector search and retrieval engines (Elasticsearch, Qdrant, Pinecone).

OUR STACK INCLUDES:

  • Container & Infrastructure: Kubernetes, Terraform, AWS GPU Infrastructure
  • ML Serving & LLM Tooling: vLLM, Ray, Triton, LiteLLM, AWS Bedrock, MLflow, SageMaker
  • Languages & Frameworks: Python, FastAPI, PyTorch
  • Data & Streaming: Kafka, Postgres, Redshift, Snowflake

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.

*Unlimited paid sick days, Wellness Days and the 90 day away program do not apply to certain roles.

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.

Source: Wealthsimple careers (Ashby)

Similar roles