Software Developer – ML/AI Compiler

Hybrid (Office in Montréal, CA)

yasp is building the next-generation AI compiler for training deep learning models, powered by LLM-based kernel generation and hardware-aware optimization. We’re looking for a passionate Software Developer to help shape the future of AI infrastructure by joining our core compiler team.

In this role, you will collaborate closely with ML specialists, compiler developers, and our LLM code generation team to accelerate model training for diverse hardware targets. You’ll play a key role in developing intelligent compiler components that translate AI models into optimized, low-level code tailored for performance.

What You’ll Do

Collaborate with ML and compiler teams to design, build, and optimize the yasp Compiler AI workload acceleration
Work with LLM-based agent systems to automate code generation and kernel optimization.
Design and implement compiler passes, runtime optimizations, and backend targeting AI workloads.
Analyze performance bottlenecks in training pipelines and apply optimizations at both framework and kernel levels.

What We’re Looking For

Strong programming skills in Python and C/C++.
Experience with ML frameworks (e.g., PyTorch, TensorFlow, ONNX, TensorRT) or compiler stacks.
Experience with compiler IRs, graph algorithms and performance optimization work.
Understanding of machine learning model internals and training workflows.
Familiarity with CUDA, LLVM, Triton, or custom codegen frameworks.
Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or equivalent practical experience.

Bonus Points

Experience working on or with ML compilers, torch.compile, MLIR, TVM, XLA, or similar.
Experience in performance tuning and low-level optimization on GPU, CPU, or AI accelerators.
Familiarity with LLM agent systems and their role in program synthesis or model compilation.
Contributions to open-source compiler, AI, or ML infrastructure projects.

‹ Research Engineer - AI for Code

Apply for this job

Why join us?

Meaningful Work

You’ll contribute directly to projects that matter — where users, clients, or communities see real benefit.

Meaningful Work

You’ll contribute directly to projects that matter — where users, clients, or communities see real benefit.

Ownership & Autonomy

We give you the space to lead, experiment, and make decisions; we don’t micromanage.

Ownership & Autonomy

We give you the space to lead, experiment, and make decisions; we don’t micromanage.

Growth Culture

You won’t be static. We support your learning with coaching, resources, and exposure to new challenges.

Growth Culture

You won’t be static. We support your learning with coaching, resources, and exposure to new challenges.

Stable & Ethical

We operate with transparency, accountability, and financial responsibility. No surprises, no hollow promises.

Stable & Ethical

We operate with transparency, accountability, and financial responsibility. No surprises, no hollow promises.

Belonging & Diversity

We believe diversity of background, thought, and experience makes us stronger. We strive for an inclusive culture where everyone can thrive.

Belonging & Diversity

We believe diversity of background, thought, and experience makes us stronger. We strive for an inclusive culture where everyone can thrive.

Long-Term Vision

We plan and invest for the long run — in the business, in our people, and in our community.

Long-Term Vision

We plan and invest for the long run — in the business, in our people, and in our community.

At yasp, we’re committed to creating an inclusive and diverse environment where everyone has an equal opportunity to thrive.

We’re looking for people who are excited by deep tech challenges, passionate about pushing the boundaries of AI performance, eager to build products that make a real impact, and absolutely thrive when working with a team.

If that sounds like you, we’d love to connect. Apply now!