At a Glance
- Tasks: Transform ML processes into cutting-edge AI solutions and build innovative infrastructure.
- Company: Join Wealthsimple, a forward-thinking tech company in Toronto.
- Benefits: Competitive salary, equity options, and a vibrant work culture.
- Other info: Exciting opportunities for growth in a dynamic, collaborative environment.
- Why this job: Be at the forefront of AI and make a significant impact in the tech world.
- Qualifications: 7+ years in software engineering with a focus on ML infrastructure and MLOps.
The predicted salary is between 152000 - 189000 $ per year.
Wealthsimple is hiring a Senior Software Developer, ML Platform & Infrastructure (Toronto Headquarters). This is an on-site AI & machine learning role listed on the official Wealthsimple careers site.
About the Role: 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.
Key Responsibilities:
- 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.
- Empower self-serve ML across Wealthsimple: Partner with product engineering and data science teams to build framework-agnostic platform tooling that abstracts infrastructure complexity.
Requirements & Details:
- 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 standards.
- 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.
- 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).
Salary: CA$152K – CA$189K • Offers Equity
Job Type: Full-time
Work Mode: On-site
Apply directly on the Wealthsimple careers page using the Apply button.
Senior Software Developer, ML Platform & Infrastructure in Toronto employer: Gomyneed Private Limited
Lyft in Toronto is an exceptional employer that fosters a collaborative and innovative work culture, empowering employees to drive impactful AI solutions. With a strong focus on professional growth, team members have access to continuous learning opportunities and the chance to work on cutting-edge technology that shapes the future of transportation. The vibrant city of Toronto offers a dynamic environment, enhancing both personal and professional experiences for those who join our team.