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