Jobs in Canada

Group jobs by employerEmail me new jobs Save this search

Added 4 months ago
C
Instrumentation & Controls Design Engineer III
CRB
Mississauga, Ontario
- $100,000 to $135,000 CAD / year
- Full-time
…people's lives. We are the leading provider of sustainable Engineering, Architecture, Construction and Consulting solutions to the global life sciences and food and beverage industries. Our…
Added 4 months ago
WI
Regional Leader – Ontario & Atlantic
WSP in Canada
Ottawa, Ontario
- Full-time
With an unmatched breadth and depth of engineering, advisory and science‑based expertise, our global minds unite to power local solutions.
Added 4 months ago
WI
Regional Leader – Ontario & Atlantic
WSP in Canada
Whitby, Ontario
- Full-time
With an unmatched breadth and depth of engineering, advisory and science‑based expertise, our global minds unite to power local solutions.
Added 4 months ago
WI
Regional Leader – Ontario & Atlantic
WSP in Canada
Mississauga, Ontario
- Full-time
With an unmatched breadth and depth of engineering, advisory and science‑based expertise, our global minds unite to power local solutions.
Added 4 months ago
WI
Regional Leader – Ontario & Atlantic
WSP in Canada
Vaughan, Ontario
- Full-time
With an unmatched breadth and depth of engineering, advisory and science‑based expertise, our global minds unite to power local solutions.
Added 4 months ago
WI
Regional Leader – Ontario & Atlantic
WSP in Canada
Toronto, Ontario
- Full-time
With an unmatched breadth and depth of engineering, advisory and science‑based expertise, our global minds unite to power local solutions.
Added 4 months ago
B
Director, AI Solutions Architect
Brookfield
Toronto, Ontario
- Full-time
…agent-based use cases. Partners with business, product, data, engineering, security, and governance teams to move high-value AI solutions from concept to…
Added 4 months ago
B
Director, Data Solutions Architect
Brookfield
Toronto, Ontario
- Full-time
…patterns. This role partners closely with business, product, engineering, and architecture stakeholders to assess requirements, define end-to-end solution designs, and ensure data…
Added 4 months ago
C
Mechanical Engineer III
CRB
Mississauga, Ontario
- $100,000 to $135,000 CAD / year
- Full-time
Intermediate experience with performing engineering calculations, and developing and implementing HVAC, mechanical utility and piping systems designs
Added 4 months ago
B
Forward Deployed Engineer
Baseten
Montreal, Quebec
- $165,000 to $330,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Software Engineer - GPU Kernels
Baseten
Montreal, Quebec
- $180,000 to $360,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Software Engineer - Billing and Internal Tooling
Baseten
Montreal, Quebec
- $165,000 to $330,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Frontend Engineer
Baseten
Montreal, Quebec
- $190,000 to $300,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Software Engineer - Infrastructure
Baseten
Montreal, Quebec
- $165,000 to $330,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Software Engineer - GPU Networking & Distributed Systems
Baseten
Montreal, Quebec
- $165,000 to $330,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Software Engineer - Enterprise Platform
Baseten
Montreal, Quebec
- $165,000 to $330,000 CAD / year
- Full-time
…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…
Added 4 months ago
B
Security Engineer
Baseten
Montreal, Quebec
- $150,000 to $250,000 CAD / year
- Full-time
Security architecture and design: Collaborate with engineering teams to design and implement secure systems and infrastructure, including cloud (AWS/GCP) environments and container…
Added 4 months ago
C
Electrical Engineer II
CRB
Mississauga, Ontario
- $80,000 to $105,000 CAD / year
- Full-time
The Electrical Engineer focuses on the design and engineering of power, lighting, grounding/bonding, special systems (e.g., telephone/ data, security, fire alarm), raceway, cable tray, and…
Added 4 months ago
D|
Parts Manager
DAG | Downtown AutoGroup
Toronto, Ontario
- Full-time
…Downtown AutoGroup (DAG), we're not just servicing cars. We're engineering trust, crafting reliability, and revolutionizing the automotive service experience. We're a community of skilled…
Added 4 months ago
H
Senior Java Cloud Developer - Landmark
Halliburton
Calgary, Alberta
- Full-time
…processing platform. This role is primarily focused on backend engineering, distributed compute systems, cloud-native services, and scalable microservice architectures. The successful…
Added 4 months ago
SI
Senior KDB+/Q Developer (Full Time) - Canada
Saransh Inc
Mississauga, Ontario
- Full-time
We are looking for a highly skilled Senior KDB Developer to join our engineering team. The ideal candidate will have strong expertise in kdb+/q, real-time data processing, time-series…
Added 4 months ago
RA
Engineering Designer, I&C
R.V. Anderson Associates Limited
Toronto, Ontario
- Full-time
We are currently seeking for an Engineering Designer, Instrumentation and Control (I&C) Group in our Toronto office. This position will provide project support to our offices across…
Added 4 months ago
M
Applied AI, Technical Lead, Forward Deployed AI Engineer
Mistral
Montreal, Quebec
- Full-time
Mistral provides full-stack AI solutions: from frontier models to developer tools, applications, and compute. We partner with enterprises tackling the hardest problems - across high-stakes…
Added 4 months ago
S
Lead Simulation Researcher (Electronics / Firmware)
Simantic
Waterloo, Ontario
- $200,000 to $300,000 USD / year
- Full-time
As the lead simulation researcher, you will be prototyping hardware systems to test and validate simulation physics to real-world behavior. You will take ownership in developing the…
Added 4 months ago
QC
Senior Quantum Scientist
Q-Block Computing
Ottawa, Ontario
- Full-time
…Q-Block Computing Research operates alongside a sustainable engineering organization whose ordinary work is the construction of ultrastable lasers, quantum-limited photodetectors, reference…

Baseten

Software Engineer - GPU Kernels

Montreal, QC, Canada

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

Apply on LinkedIn About employer

Applications open in a new tab.

Job details

Pay: $180,000 to $360,000 CAD / year
Job type: Full-time
Posting language: English
Posted: Jul 10, 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 to ship AI products.

THE ROLE

We’re seeking a GPU Kernel Engineer to join our team at the cutting edge of AI acceleration, where your code directly impacts the performance of state-of-the-art machine learning models. As a GPU Kernel Engineer, you'll craft the foundation that powers modern AI workloads, optimizing every microsecond of computation to enable breakthrough applications.

You'll work in a fast-paced, intellectually stimulating environment where technical excellence is paramount and your contributions directly influence production systems serving millions of users across numerous products. This role offers exceptional growth potential for engineers passionate about low-level optimization and high-impact systems work.

EXAMPLE INITIATIVES

You'll get to work on these types of projects as part of our Model Performance team:

Baseten Embeddings Inference: The fastest embeddings solution available
The Baseten Inference Stack
Driving model performance optimization

Responsibilities

Core Engineering Responsibilities

Design and implement high-performance GPU kernels for key ML operations, including matrix multiplications, attention mechanisms, and mixture-of-experts routing
Write and optimize code using CUDA, PTX assembly, and architecture-specific techniques
Apply advanced performance optimization methods such as memory coalescing, warp-level programming, tensor core acceleration, and compute/memory overlap

Performance & Innovation

Implement cutting-edge features like quantization (FP8/FP4), sparsity, and compute/communication overlap
Identify and resolve performance bottlenecks using tools like Nsight Systems, Nsight Compute, and Torch Profiler
Collaborate with research teams to productionize theoretical advancements

Impact & Collaboration

Contribute to internal and open-source GPU libraries
Present technical contributions at industry conferences (e.g., NVIDIA GTC, AWS re:Invent)

Requirements

Strong understanding of GPU architecture and programming paradigms:
- Memory hierarchy (global, shared, registers, L1/L2 cache)
- Thread/block/grid organization
- Synchronization techniques and race condition mitigation
Proficient in C++ and GPU performance profiling tools
Knowledge of:
- CUDA C++ API
- Memory access patterns and bandwidth optimization
- Numerical precision and quantization strategies
- Modern GPU features (e.g., tensor cores, async operations)

NICE TO HAVE

Experience with Transformer models and attention optimization (e.g., Flash Attention)
Familiarity with GPU kernel libraries: Cutlass, Triton, Thrust, CUB
Background in GEMM tuning and distributed/multi-GPU compute
Contributions to open-source GPU projects
Research publications or conference presentations on GPU performance

Benefits

Competitive compensation, including meaningful equity.
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
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 on LinkedIn About employer

More about this job

Responsibilities

Design and implement high-performance GPU kernels for key machine learning operations. Collaborate with research teams to optimize performance and contribute to internal and open-source GPU libraries.

Requirements

Strong understanding of GPU architecture and programming paradigms is required. Proficiency in C++ and knowledge of CUDA and GPU performance profiling tools are essential.

Skills

GPU Architecture
CUDA
C++
Performance Profiling
Memory Optimization
Quantization
Attention Mechanisms
Matrix Multiplications
Tensor Core Acceleration
Sparsity
Compute/Communication Overlap
Synchronization Techniques
Race Condition Mitigation
GEMM Tuning
Distributed Compute
Open-Source Contributions

Benefits

Health Insurance
Stock Options
401k
Flexible PTO
Paid Parental Leave
Fertility and Family-Building Stipend

Visa sponsorship

Not detected in the job text

Keywords

GPU
CUDA
C++
Machine Learning
Performance Optimization
Quantization
Attention Mechanisms
Matrix Multiplications
Tensor Cores
Sparsity
Compute Overlap
Profiling Tools
Memory Hierarchy
Thread Organization
Race Conditions
GEMM Tuning
Open-Source
AI
Inference
Model Performance
NVIDIA
AWS
Cutlass
Triton
Thrust
CUB
Flash Attention
Nsight Systems
Nsight Compute
Torch Profiler

Report this job

Work requirements

Jobs in Canada