Intern Researcher Agentic ML Compiler

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

Tasks

• 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

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

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.

Ready to apply Submit your application through this posting and our team will review it personally. From there you can expect a short intro call a technical evaluation built around real problems and conversations with the people youd work with directly.