Software Developer – ML/AI Compiler (Montreal)

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 - ML/AI Compiler 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. 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. Growth Culture You won’t be static. We support your learning with coaching, resources, and exposure to new challenges. Secure & 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. 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! Apply on Kit Job: kitjob.ca/job/2paj78