Find work in Canada.

555,662 jobs across our Canadian network. How we count

Jobs in Canada

Email me new jobs Save this search

Added 26 days ago
P&
Architecte de données / Data Architect
Pratt & Whitney
Longueuil, QC
- Full-time
Added 26 days ago
SP
Professor, Part-Time
Seneca Polytechnic
Toronto, ON · North York, ON
- Part-time
Professional Engineer (P.Eng.) designation is considered a strong asset, particularly for candidates with experience applying AI in engineering or technology environments.
Added 26 days ago
AL
Principal Product Applications Engineer - Scorpio
Astera Labs
Vancouver, BC
- Full-time
…accelerators at rack scale. As a Principal Product Applications Engineer on the Scorpio team, you will be the technical face of our switching portfolio to the world's leading hyperscalers,…
Added 26 days ago
T
Senior Data Engineer
TekRek
Montreal, QC
- $140,000 to $160,000 CAD / year
- Full-time
As a Senior Data Engineer, you'll own data platform design and delivery for client engagements, from scoping and architecture through to production. You'll work directly with client…
Added 26 days ago
BC
Senior Manager, Data & AI
BDO Canada
Vancouver, BC
- Full-time
Previous hands-on experience as a technical delivery professional, architect, developer, data engineer, analytics leader, or similar role, with the ability to remain technically credible…
Added 26 days ago
CA
Brand Engineer
Calico AI
Toronto, ON
- Full-time
…of it, is a hands-on job. That is why we are hiring a Brand Engineer: someone who sits with the customer, maps how they work today, and builds the path from first pilot to running their…
Added 26 days ago
Senior Staff Software Engineer
Diligent Corporation
Vancouver, BC
As a Senior Staff Software Engineer on Diligent's Platform AI team, based out of our technical center of excellence in Vancouver, you will drive the architecture and technical direction of…
Added 26 days ago
Dynamics/Powerplatform engineer
Capgemini
Montreal, QC
We are looking for a Senior Microsoft power platform Engineer, who is familiar with AI Builder and Co-Pilots as well as Dynamics and Power Apps. The ideal candidate will have experience…
Added 26 days ago
YI
AI Engineer
YuJa Inc.
Toronto, ON · North York, Ontario
- Full-time
…inclusive, user-centric solutions. We’re looking for a passionate AI Engineer to join our core engineering team and help design, build, and scale our AI infrastructure and…
Added 26 days ago
Senior Machine Learning Engineer, Recommendations
Lyft
Toronto, ON
We are seeking a Senior Machine Learning Engineer to join the Rider Applied AI team and lead the design, development, and deployment of state-of-the-art machine learning and artificial…
Added 26 days ago
Senior Software Engineer, Full-Stack Applications - Python
Fitch Group
Toronto, ON
Senior Software Engineer, Full-Stack Applications – Python
Added 26 days ago
R
Architecte de données / Data Architect
RTX
Longueuil, QC
- Full-time
Added 26 days ago
Sr Software Dev Engineer III, RDS for Oracle
Amazon
Vancouver, BC
Amazon Web Services is hiring a Software Development Engineer III for the RDS for Oracle team. We are looking for an experienced engineer to design and build the next generation of managed…
Added 26 days ago
Senior Data Scientist I
Bell
Toronto, ON
The Cloud Capability: Extensive hands-on knowledge of Google Cloud Platform (GCP); holding a GCP Professional Machine Learning Engineer certification is a massive bonus!
Added 26 days ago
Senior Software Engineer, Data Platform
Okta
Toronto, ON
Added 27 days ago
M
AI Engineer
maru.ai
Engineer, BC
- Full-time
We are an early stage startup developing cutting-edge AI technology. We are looking for a talented AI engineer to join our small but growing team. You will have the opportunity to take on a…
Added 27 days ago
M
Software QA Engineer
maru.ai
Engineer, BC
- Full-time
We are looking for a detail-oriented Software QA Engineer to join our fast-growing AI/ML startup. In this role, you will help ensure the quality of our AI/ML products by designing and…
Added 27 days ago
BS
ML/AI Engineer
Brock Software
Engineer, BC
- Full-time
We're seeking an experienced engineer to fine-tune open source models, train them on custom datasets, and deploy them to production using GPU infrastructure on platforms like fly.io.
Added 27 days ago
J
Machine Learning Engineer Intern
Jobright.ai
Canada
- Internship
…helps users reach better job outcomes. The Machine Learning Engineer Intern will develop and evaluate machine-learning features, support data pipelines and integrations, investigate model…
Added 27 days ago
E
Principal Product Manager - Identity Threat Detection and Response
Elastic
Canada
- Full-time
…You are an advocate for the security analyst and detection engineer, and you are focused…
Added 27 days ago
M
Security Engineer II - Identity and Access Management
Marqeta
Remote, Canada
- Full-time
- Remote
We’re seeking an experienced Security Engineer II with a strong passion for Identity and Access Management (IAM) and expertise in cloud-native environments, particularly AWS. You'll build,…
Added 27 days ago
B&
Lead, AI Engineering
Bain & Company
Zurich, ON
- Full-time
As an AI Engineer in AIS, you will build the technical core of these transformations and work as part of broader Bain consulting teams to move solutions from prototype to real adoption. The…
Added 27 days ago
G
Browser Security Engineer Intern, (Winter 2026)
GeoComply
Vancouver, BC
- Internship
Added 27 days ago
B
Software Engineer- Inference Performance
Baseten
Montreal, QC
- $180,000 to $360,000 USD / year
- Full-time
You'll get to work on these types of projects as an Inference Performance engineer:
Added 27 days ago
B
Software Engineer- Inference Platform
Baseten
Montreal, QC
- $180,000 to $360,000 USD / year
- Full-time

Baseten

Software Engineer- Inference Performance

Montreal, QC, Canada

$180,000 to $360,000 USD / year · Full-time

Apply About employer

Applications open in a new tab.

Job details

Pay: $180,000 to $360,000 USD / year
Job type: Full-time
Apply by: Nov 4, 2026
Posting language: English
Posted: Oct 5, 2026

Original job description

About Baseten

Baseten powers mission-critical inference for the world's most dynamic AI companies, like Cursor, Notion, OpenEvidence, Abridge, Clay, Gamma, and Writer. By uniting applied AI research, flexible infrastructure, and seamless developer tooling, we enable companies operating at the frontier of AI to bring cutting-edge models into production. We're growing quickly and recently raised our $1.5B Series F, led by Altimeter Capital, Conviction Partners, and Spark Capital. Join us and help build the platform engineers turn to ship AI products.

THE ROLE

We're looking for inference performance engineers who want to make the world's most demanding AI workloads run faster and more efficiently. You'll work across the stack, from the inference engine and runtime through scheduling, serving, and routing. Along the way you'll apply techniques like prefill/decode disaggregation, speculative decoding, and KV-cache management. You'll reason from first principles about where time and memory go, find what's holding performance back, and close the gap. Your work directly impacts how fast our customers' models run and how efficiently we serve them. This role is ideal for someone who thrives in a fast-paced startup environment and is eager to make significant contributions to the exciting field of LLM inference.

EXAMPLE INITIATIVES

You'll get to work on these types of projects as an Inference Performance engineer:

Agentic Kernels in Production
How we built the new fastest API for GLM-5.2
Live draft model training for speculative decoding
Making Kimi K3 Tokenization 18x faster
The Baseten Inference Stack
Driving model performance optimization

Responsibilities

Implement and productionize cutting-edge inference techniques, working deep in runtime internals. That includes quantization, speculative decoding, KV-cache reuse, chunked prefill, LoRA, guided generation for structured outputs, and custom scheduling and routing algorithms.
Profile and optimize inference end to end, from kernel launch overhead and memory layout up to request scheduling, prefill/decode disaggregation, and cache-aware routing. Run cross-layer investigations, such as tracing a tail-latency regression from request timing through routing and batching down to a kernel.
Turn performance into cost savings. Improve tokens per GPU-hour, raise utilization, and give customers and internal teams clear latency/throughput/cost tradeoffs.
Bring up and tune new model architectures on new hardware quickly, often in the same week they're released.
Build benchmarking frameworks that measure real-world performance across model architectures, batch sizes, sequence lengths, and hardware configurations.
Contribute upstream to open-source inference engines (vLLM, SGLang, TensorRT-LLM), and partner closely with model, infrastructure, and customer-facing teams to ship wins.

Requirements

Bachelor's, Master's, or Ph.D. degree in Computer Science, Engineering, Mathematics, or related field.
Experience with one or more general-purpose programming languages, such as Python or C++.
Familiarity with LLM optimization techniques (e.g., quantization, speculative decoding, continuous batching).
Strong familiarity with ML libraries, especially PyTorch, TensorRT, or TensorRT-LLM.
Demonstrated interest and experience in LLMs.
Deep understanding of GPU architecture.

NICE TO HAVE

Proficiency in enhancing the performance of software systems, particularly in the context of large language models (LLMs)
Contributed to vLLM, SGLang, TensorRT-LLM, or another inference engine.
Worked on large-scale distributed serving: autoscaling, load balancing, multi-region or multi-cloud capacity.
Written or optimized GPU kernels (CUDA, Triton, CUTLASS, or similar)
Worked on quantization (FP8/FP4, AWQ, GPTQ) or speculative decoding in production.
Deep understanding of software engineering principles and a proven track record of developing and deploying AI/ML inference solutions.

Benefits

Competitive compensation, including meaningful equity
(U.S. only) 100% coverage of medical, dental, and vision insurance for employee and dependents
Flexible PTO policy including company wide Winter Break (our offices are closed from Christmas Eve to New Year's Day!)
Paid parental leave
Fertility and family-building stipend through Carrot
(U.S. only) Company-facilitated 401(k)
Exposure to a variety of ML startups, offering unparalleled learning and networking opportunities.

Apply now to embark on a rewarding journey in shaping the future of AI! If you are a motivated individual with a passion for machine learning and a desire to be part of a collaborative and forward-thinking team, we would love to hear from you.

At Baseten, we are committed to fostering a diverse and inclusive workplace. We provide equal employment opportunities to all employees and applicants without regard to race, color, religion, gender, sexual orientation, gender identity or expression, national origin, age, genetic information, disability, or veteran status.

We are an Equal Opportunity Employer and will consider qualified applicants with criminal histories in a manner consistent with applicable law (by example, the requirements of the San Francisco Fair Chance Ordinance, where applicable).

Compensation Range: $180K - $360K

Apply About employer

More about this job

Responsibilities

Implement and optimize production LLM inference across runtimes, kernels, scheduling, serving, and routing, using techniques such as quantization, speculative decoding, and KV-cache management. Improve latency, throughput, GPU utilization, and cost; benchmark new models and hardware; and contribute to open-source inference engines while collaborating across teams.

Requirements

A bachelor's, master's, or Ph.D. degree in computer science, engineering, mathematics, or a related field is required, along with experience in a general-purpose language such as Python or C++. Candidates should have LLM optimization experience or familiarity, strong knowledge of ML libraries such as PyTorch or TensorRT, demonstrated interest and experience in LLMs, and a deep understanding of GPU architecture.

Skills

LLM Inference Optimization
Python
C++
PyTorch
TensorRT
GPU Architecture
Quantization
Speculative Decoding
Continuous Batching
KV-Cache Management
Performance Profiling
GPU Kernel Optimization
Distributed Model Serving
Benchmarking
Scheduling and Routing
Inference Engine Development

Benefits

Competitive Compensation
Equity
Medical Insurance
Dental Insurance
Vision Insurance
Flexible Paid Time Off
Company-Wide Winter Break
Paid Parental Leave
Fertility and Family-Building Stipend
401(k)
Learning and Networking Opportunities

Visa sponsorship

Not detected in the job text

Keywords

LLM Inference
Inference Performance
Python
C++
PyTorch
TensorRT
TensorRT-LLM
vLLM
SGLang
CUDA
Triton
CUTLASS
GPU Architecture
Quantization
FP8
FP4
AWQ
GPTQ
Speculative Decoding
Continuous Batching
KV-Cache
Chunked Prefill
LoRA
Guided Generation
Prefill/Decode Disaggregation
Distributed Serving
Autoscaling
Load Balancing
Benchmarking
Performance Optimization

Report this job