Intern Researcher - Agentic ML Compiler

yasp.ai GmbH•Montreal, QC

18h

About The Position

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.

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.

Nice To Haves

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).

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.