Intern Researcher - Agentic ML Compiler

Hybrid (Montréal, CA)

Location: Hybrid, Montreal Canada

About Us:

yasp is pioneering the future of software development with a compiler that leverages agentic AI for advanced optimization and code generation. We are looking for an Intern Researcher - Agentic ML Compiler to join our team and help drive the core innovation that will define the next generation of our technology.

We don’t draw boundaries between research and engineering. We are creative thinkers and relentless prototypers who live at the bleeding edge of AI research and engineering. Together, we explore, invent, and ship novel technology that pushes beyond the state of the art for accelerating deep learning models.

Responsibilities:

Develop novel agentic AI and LLM-based methods to optimize kernels for machine learning models.
Evaluate the performance of AI driven ML compilation techniques via rigorous benchmarking.
Stay up to date with the publication landscape, open-source projects, and industry trends.
Contribute to writing research papers for publication in top AI conferences.
Work with a talented team towards breakthroughs in AI for ML compilation.

Requirements:

Master’s or PhD student in Computer Science/Maths/Physics with a focus on AI/ML, or equivalent practical experience.
Exceptional prior work in AI/ML, ideally involving LLMs or agents to solve complex tasks.
Strong engineering and coding abilities (PyTorch, agentic AI frameworks, LLM post-training frameworks, software development best practices).
Passion for good scientific practice, blending clean formal thinking and good communication with exacting engineering.

Preferred:

Experience post-training LLMs using SFT and RL, especially as part of an AI agent.
Knowledge of ML compiler principles including intermediate representations, operator fusion, and custom CUDA or Triton kernel generation.
Previous publications in leading AI conferences (e.g., NeurIPS, ICLR, ICML).

Perks and Benefits:

Competitive salary.
Comprehensive health benefits (including dental).
Opportunities for professional development and growth.
Flexible work hours.
Dynamic and collaborative work environment.
Cutting-edge software and hardware platforms.

‹ General Application

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!